1 Introduction

Lab book for analyses using hierachal computational modelling to identify paramters that define the best model of learning as it applies to fear conditioning acquisition and extinction using FLARe fear conditioning data. Long abstract, justification and analysis plan found in prelim manuscript here

In short:

1.1 Aims

  1. Identify model of learning based on a priori hypotheses that best fits the trajectories of fear relevant learning in our FLARe dataset
    • Use all first week data from Validation, app TRT, lab TRT, Pilot, Headphones (n = 223 after exclusions)
    • Include Acquisition, extinction (trajectories representing fear learning and treatment)
    • Identify parameters that define these trajectories
      • e.g. Learnign rate, plateau, first ambiguous trial etc.
  2. Cross validate best fitting model in TEDS data

  3. Are these parameters associated with other emasures of indsividual differences in our datasets?
    • Personality (Neuroticism)
    • Current anxiety symptoms (GAD-7) - equivalent of baseline symptoms (Chris + Meg analyses)
    • Lifetime / trait anxiety (STAI / ASI - FLARe analyses)
    • Current depression symptoms (PHQ-9) - equivalent of baseline symptoms (Chris + Meg analyses)
    • Interpretation biases (IUS, ASSIQ - FLARe analyses)
    • SES (Meg IAPT: benefits, employment etc)
    • Gender (Meg analyses)
    • Emotion regulation profile (potentially LCA based?)

1.2 Impact and relevance

Evidence from both human (Richter et al., 2012) and rodent (Galatzer-Levy, Bonanno, Bush, & LeDoux, 2013) studies suggest that trajectories of how we learn and extinguish fear differ between individuals. Different trajectories of fear and extinction have also been found using fear conditioning studies (e.g. Duits et al., 2016), a good model for the learning of, and treatment for, fear and anxiety disorders. It is likely that these trajectories of fear extinction might predict outcomes in exposure-based cognitive behavioural therapy (Kindt, 2014). 
 
Identifying parameters that predict individual trajectories of fear learning and extinction will enable us to harness fear conditioning data more effectively to aid in understanding mechanisms underlying the development of and treatment for anxiety disorders. With more accurate models of these processes, the potential to use fear conditioning paradigms to predict those most at risk of developing an anxiety disorder, and those who might respond best to exposure-based treatments, greatly improves.

1.3 Useful references

Sutton and Barto Reinforcement Learning - Textbook on reinforcement learning
Anxiety promotes memory for mood-congruent faces but does not alter loss aversion (Charpentier…Robinson, 2015) - Good example of a sensitivity learning parameter
Hypotheses About the Relationship of Cognition With Psychopathology Should be Tested by Embedding Them Into Empirical Priors (Moutoussist et al., 2018) - Including variables of interest (e.g. anxiety) in the model

Toby Wise has just submitted an aversive learning paper incorporating beta probability distributions in the best model for uncertain learning paramters etc.

A copy of this is

1.4 Analysis plan

  1. Define set of a priori models moving from simple to more complex
    • Some paramters to include:
      • Rate of learning (sometimes with punishment reinforcement)
      • Sensitivity to punishment
      • Pre-existing anxiety
      • SES? Gender?
  2. Run each model and compare fit in FLARe pre TEDS data
    • Use Log likelihood and BIC etc.
  3. Select best fitting model

  4. Extract individual data for learning parameters from this model and see what factors best predict it
    • Anxiety (if anxiety isnt best as part of the model)
    • Interpretation biases
    • Tolerance of uncertanty
    • Cognitive emotional control
    • emotional attentional control
    • SES?
    • Gender?
  5. Run all models again in FLARe TEDS
    • Decide if the same model best fits the data again.
    • See if we get similar results from the parameter prediction

Will use a combination of R.Version(3.5.1), RStan (Version 2.18.2, GitRev: 2e1f913d3ca3) and hBayesDM package in R (3.5.1) Ahn, W.-Y., Haines, N., & Zhang, L. (2017). Revealing neuro-computational mechanisms of reinforcement learning and decision-making with the hBayesDM package. Computational Psychiatry, 1, 24-57., which uses RStan

1.5 Modelling notes

1.5.1 Intuition

Discussion with Vince Valton and Alex Pike about the best way to fit this model. As the observed outcomes (expectancy ratings) are non binary and are related to eachother (i.e. as you become more likely to select 9, you become less likely to select 1) we should consider each trial for each person for each stimulus as a constantly updating beta distribution. so you might see a pattern like this for the CS+ in acq for example.

So, best model is likely to be one using beta distributions that show the probability distribution for each rating.

We can use sufficient parameters to describe these (i.e. mean / sd or possibly the mode)

A useful intuition of the beta distribtion can be found here

and a useful website here

scaling

We can scale the beta by how aversive participants find the shock. i.e. it might update their learning as if there was .5 a shock or 1.5 of a shock depending on their own sensitivity to the aversiveness / punishment.

alpha

generalisation

We can do this with a single beta distribution for each phase (collapsing over the two stimuli). This would be akin to a per phase generalisation paramaterer in that it will be smaller if they tend to choose the same expectancy for both stimuli and larger if they tend to choose very differently for both stimuli.

However, because these variables are not really equivalent (i.e the reinforcement rate is different for both, and we use this in the model)

So instead we can create a paramater which is the value of cs- weighted by some value of the cs+. How much each individual weights by the Cs+ can be freely estimated by the model and can be the generalisation paramter.

So this would be vminus = vminus + (w)vplus (where the w paramter is the freely estimated paramter per person)

per stimulus We probably want to model cs+ and cs- separately too - so have a beta distribution characterised by sufficient parameters for each.

per trial

All of the above can then also be done with updating per trial.

leaky beta

we also need a model that incorporates ‘leak’. i.e. learning leak - likely that participants will update more based on more recent trials and learn less from the more distant trials as time progresses. See Toby’s paper for more.

uncertainty

We should consider incorportating a paramter that maps to participant uncertainty about outcomes.

anxiety

Might be worth incorporating this as a model paramater / feature. Read this for more.

Hypotheses About the Relationship of Cognition With Psychopathology Should be Tested by Embedding Them Into Empirical Priors (Moutoussist et al., 2018)

1.5.2 Log likelihood notes

As we are using a beta distribution, we will calculate log likelihood based on the probability function for the distribution (i.e. where will the peak of the shape be) given the participants response at each trial. So will add the probability density function given each trial response trial by trial for each of the CS+ and - summed together.

Will obtain 1 log likelihood and then 1 per trial and add together to make sure that these are comparable.

the basic stan terminology for this is below:

beta_lpdf(rating[t,p]|shape1[t,p],shape2[t,p])

where beta_lpdf is the probability density given the rating made and each of the two beta distribution shape paramters that we estimate.

This is what we will use to compare models.

1.5.3 Terminology

V == ‘value’. Baasically a paramter that is about the salience of the stimulus at any given point.
alpha == ‘learning rate’. A parameter that describes how sensitive people are to updating their learning. So a fast learning rate means that learning on any given trial is weighted more based on the trials immediatly preceding than past ones, and a slow learning rate means that all past events influence learning more evenly. Alex’s tennis analogy is good here (Federer - stable player, can predict a win based on all matches; Murray - volatile player; his last match is best predictor of next match performance). beta == ‘confidence’. This is sort of an error term - how much variance in rating choice is there for each person/trial. Can be thought of as the variance, or beta^2 as the sd.

Can be confusing as we are using beta distributions (different thing) which has two sufficient parameters a + b).

1.5.4 Beta distribution visualisation

and how they change depending on whether you change the beta or alpha paramters.

A really nice summary visualisation

A really nice summary visualisation

Here are some simulations I can change and play with the illustrate the same sort of thing.

##  [1] 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
##  [1] 0.00 0.05 0.10 0.15 0.20 0.25 0.30 0.35 0.40 0.45 0.50 0.55 0.60 0.65
## [15] 0.70 0.75 0.80 0.85 0.90 0.95 1.00
## [1] "stable beta, increasing alpha"

## [1] "stable alpha, increasing beta"

1.5.5 Models to write / run

Will probably do all per trial. Will do an early sensitivity check to confirm this.

  1. Single beta, no scaling
  2. Single beta, no scaling per trial.
    *** At this point, compare the two above. Ensure the per trial fits better, and if it does then do all below per trial***
  3. “” scaled
  4. Single beta Single alpha reinforcement learning model (estimate both the beta and the alpha i.e. learning rate)
  5. Single beta single alpha reinforcement learning with mean + sd for the beta estimate as a paramter
  6. Beta per stimulus
  7. Beta per stimulus + generalisation paramter (Vminus = vminus + wvplus)
  8. Leaky beta
  9. Leaky beta + uncertainty
  10. Leaky beta + uncertainty + anxiety

1.5.6 Justification of model components

Alpha Learning rate paramter. If high then will be very influenced by previous trial events, if low, then will be more standardly influenced by accumulating events.

  • Single alpha per person
    • Assumes that learning rate is a constant for each individual that might be scaled by other factors, such as certainty or sensitivity.

Betas Variance/certainty parameter

  • Single beta per person
    • Assumes that the general variance around ratings is the same regardless of stimulus. i.e. as much uncertainty for CS+ asn CS-
  • Two beta’s per person
    • Assumes that confidence / uncertainty might differ by stimulus. Presumably as a factor of reinforcement rate.

2 Analyses

2.1 Set up stan

These use Alex Pikes RStan script with minor modification to make it punishment only to see if it runs. Testing that the approach works with the current data set up etc.

The settings for the script are below, including stan chain paramters and directory set up.

## Warning: package 'tibble' was built under R version 3.5.2

This loads the libraries and source files needed to run this script, and sets up RStan

## Warning: package 'dplyr' was built under R version 3.5.2
## Warning: package 'StanHeaders' was built under R version 3.5.2
## Warning: package 'data.table' was built under R version 3.5.2

2.2 Preliminary

2.2.1 Compare a priori to data

2.2.1.1 Simulate different learning rates

only doing this ‘accurately’ for the acquisition CS+, as the simulations require probability. I am using contingency for this (0.75). If set for 0 for all other phases and stimuli then it looks as if the learning should be flat regardless of alpha. We expect in reality that this probability will vary between people and will be unlikely to be zero. So test 12 and 18 trials with a probability of 0.5 and 0.2 as well.

2.2.1.1.1 12 trials; probability = 0.75
## [1] "Simulated learning rates. 12 trials; probability = 0.75 (CSp acq contingency) \n"

##### 12 trials; probability = 0.5

## [1] "Simulated learning rates. 12 trials; Probability = 0.5\n"

2.2.1.1.2 12 trials; probability = 0.2
## [1] "Simulated learning rates. 12 trials; Probability = 0.2\n"

2.2.1.1.3 18 trials; probability = 0.5
## [1] "Simulated learning rates. 18 trials; Probability = 0.5\n"

2.2.1.1.4 18 trials; probability = 0.2
## [1] "Simulated learning rates. 18 trials; Probability = 0.2\n"

2.2.1.2 Plot subset of trajectories in flare

## 
## Attaching package: 'reshape2'
## The following objects are masked from 'package:data.table':
## 
##     dcast, melt
## The following object is masked from 'package:tidyr':
## 
##     smiths
## The following objects are masked from 'package:reshape':
## 
##     colsplit, melt, recast

2.2.2 Try RStan

See if the basic punishment only learning model for the CS+ and CS- works with the FLARe master data

2.2.2.1 Run the 8schools check

From the rstan github

This is to check that all is compiling and working and to give and idea of data format etc.

2.2.2.2 Adjust dataframe

load in the week 1 app and lab data for FLARe pilot, TRT and headphones studies. Make it long form.

Try with acquisition data first. This is formatted with no column names, with no missing data.

Derive the n parameter for both files and check these match

stanname='punish_only.stan'
minus_name <- 'bayes_acq_minus.csv'
plus_name <- "bayes_acq_plus.csv"

stanfile <- file.path(scriptdir, stanname)
minusfile <- file.path(datadir,minus_name)
plusfile <- file.path(datadir,plus_name)


minus <- fread(minusfile,data.table=F)
plus <-fread(plusfile,data.table=F)

nacqm <- dim(minus)[1]
nacqp <- dim(plus)[1]


## check that these match and create nsub variable for RStan

if (nacqm == nacqp) {
  print('subject number match')
  nsub <- nacqm
  
  print(paste('nsub set to',nsub,sep=" "))
} else {
  print('WARNING: subject number does not match. Check master dataset')
}
## [1] "subject number match"
## [1] "nsub set to 342"
# check the file format is ok

minus[1:2,]
plus[1:2,]

The expectancy rating datasets look like they are formatted fine and ntrials and nsub variables should exist.

2.2.2.2.1 Create scream data

Need to go back to stage zero and keep scream yes/no as a variable. For now to see if this runs create simulated version for the CS+. CS- will remain the same.

screamMinus <- matrix(0L,nrow=nsub, ncol=ntrials)

# Initialise plus dataset in the same way, but make the first trial 1 for everyone, then add 8 additional random 1's per person. Do this in four random patterns to mimic the real data

sc1 <- c(1,1,0,1,0,0,1,1,1,1,1)
sc2 <- c(0,1,1,1,0,0,1,1,1,1,1)
sc3 <- c(1,1,1,0,1,0,1,0,1,1,1)
sc4 <- c(1,0,1,1,0,0,1,1,1,1,1)


screamPlus <- matrix(0L,nrow=nsub, ncol=ntrials)

screamPlus[,1] <- 1


# for (n in 1:dim(screamPlus)[1]) {
#   print(n)
#   screamPlus[n,2:12] <- sample(patts,1,replace=T)
# }


for (n in 1:dim(screamPlus)[1]) {
  
  a <- sample(c(1,4),1)
  
  if (a == 1) {
    screamPlus[n,2:12] <- sc1
  } else if (a == 2) {
    screamPlus[n,2:12] <- sc2
  } else if (a == 3){
    screamPlus[n,2:12] <- sc3
  } else {
    screamPlus[n,2:12] <- sc4
  }
}
2.2.2.2.2 make rating data binary

for now to see if stan runs using bernoulli-logit function make binary resposnes from expectancy i.e. >=4.5 ==1, <= 4.5 ==0.

binarise <- function(x) {
  ifelse(x >= 4.5,1,0)
}
minusb <- data.frame(apply(minus,2,function(x) binarise(x)))

plusb <- data.frame(apply(plus,2,function(x) binarise(x)))

2.2.2.3 Set up procedure to create and sync models.

This directs to my local machine here /Users/kirstin/Dropbox/SGDP/FLARe/FLARe_MASTER/Projects/Hierachal_modelling/Scripts and is remotely linked to the github repository here.

2.2.2.3.1 Make sure the most up to date stan file is in the remote repo

git pull Bayes_modelling
   
## Already up to date.
2.2.2.3.2 check existing paramters

Unhash this if you want to check what the model looks like within the notebook.

stanname="punish_only.stan"
scriptdir="/Users/kirstin/Dropbox/SGDP/FLARe/FLARe_MASTER/Projects/Hierachal_modelling/Scripts"

#cat $scriptdir/$stanname
2.2.2.3.3 Make any changes

use echo to push these to the new file if you want to make changes from here.


## initialise bash directory and filename

stanname="punish_only.stan"
scriptdir="/Users/kirstin/Dropbox/SGDP/FLARe/FLARe_MASTER/Projects/Hierachal_modelling/Scripts"


#echo "<any changes here>" > $scriptdir/$stanname

2.2.2.4 Now try run stan

unhash this to run experimental script that checked if stan runs. This was mostly to check data formatting and installation / compilation etc.

flare_data<-list(ntrials=ntrials,nsub=nsub,includeTrial = rep(1,ntrials), screamPlus=t(screamPlus),screamMinus=t(screamMinus),
                 ratingPlus=t(plusb),ratingMinus=t(minusb))

#flare_fit <- stan(file = stanfile, data = flare_data, iter=chain_iter, chains = chain_n) #add working dir?

#save(flare_fit, file=file.path(datadir,'flare_fit_test'))

#traceplot(flare_fit,'lp__')

# extract fit data
#summary_flare<- summary(flare_fit)

# extract model summary data

#flare_loglike<- extract_log_lik(flare_fit, parameter_name = "loglik", merge_chains = TRUE)

view the fit information

#summary_flare

extract the loglikelihood using loo

#loo(flare_loglike)

so, good news is this all works. So preliminary check a success. Next need to consider the appropriate model.

2.3 Create stan friendly datasets

2.3.1 notes

We need to rescale our dataset here to be between 0 and 1.

Importantly, because we are using the proportion of trials that are not reinforced as a known paramter for statistical reasons (we don’t want a proportion of .75 and 1, better to have .25 and 0), we have made our rescaled expectancy values as 1 - rescaled(x). This means that we will still be able to interpret the results in the expected way (i.e. higher rating is greater expectation of the outcome).

2.3.2 rescale data

rescale the 1-9 expectancy values to be on a 0-1 scale.

stan cannot deal with the extreme limit of the beta, so make the rescaled limits just above 0 and below one

Note that when a value had to be imputed as it was missing it will not be an integer. Thus the function needs to allow for ranges between values.

library(scales)

# rescale and flip so that we are effectively rating the expectation that they WILL NOT hear a scream to match stan

## rescaling such that the distribution spaces the numbers 1-9 evenly. the first interval upper bound would be 0.11, then 0.22 etc. this means that the mid point of each itnerval will be:

print("mid point of each evenly spaced interval representing values between 1-9")
## [1] "mid point of each evenly spaced interval representing values between 1-9"
seq(0.5/9,1,1/9)
## [1] 0.05555556 0.16666667 0.27777778 0.38888889 0.50000000 0.61111111
## [7] 0.72222222 0.83333333 0.94444444
## thus 1 will be 1-0.055 etc.

## NOTE: might want to consider making this more flexible. enter in the numer of choice options as a variable - would be very easy. add to function library at later stage

scale_flare <- function(x){
  
  vals <- seq(0.5/9,1,1/9)
  
  for (val in 1:9){
    if (x > val-1 & x <= val){
      x <- 1 - vals[val]
    }
  }
  return(x)
}


## initialise minus_scaled dataframe.

minus_scaled <- data.frame(matrix(ncol=dim(minus)[2],nrow = dim(minus)[1]))

##  populate with rexcaled values

for (sub in 1:dim(minus)[1]){
  for (col in 1:dim(minus)[2]){
    
    minus_scaled[sub,col] <- scale_flare(minus[sub,col])
  }
}

## ditto for plus

plus_scaled <- data.frame(matrix(ncol=dim(minus)[2],nrow = dim(minus)[1]))

for (sub in 1:dim(plus)[1]){
  for (col in 1:dim(plus)[2]){
    
    plus_scaled[sub,col] <- scale_flare(plus[sub,col])
  }
}



## this is the number that will take from the midpoint to the top and bottom for the new boundaries (with ratings representing the midpoint)

cdf_scale <- 1/18

2.3.3 create proportion screams data

This is a vector containing the absolute number of trials where no scream occurred for each stimulus. As there was a 75% reinforcement rate for the CS+ (9/12 trials), this is a vector of ’3’s. For the CS-, no trials were reinforced so is a vector of ’12’s

No_scream_p <- rep(3,nsub)
No_scream_m <- rep(12,nsub)

2.3.3.1 Create scream data

Create datasets for the acquisition CS- and extinction CS+ and CS- reflecting that no screams occurred at all. Then use the pattern id variable to create a dataset for the acquisition CS+ indicating when a scream occurred for each participant.

## Create the no scream daatsets for all

screamMinus <- matrix(0L,nrow=nsub, ncol=ntrials)

# Initialise plus dataset in the same way, but make the first trial 1 for everyone, then add 8 additional random 1's per person. Do this in four random patterns to mimic the real data

sc1 <- c(1,1,0,1,0,0,1,1,1,1,1)
sc2 <- c(0,1,1,1,0,0,1,1,1,1,1)
sc3 <- c(1,1,1,0,1,0,1,0,1,1,1)
sc4 <- c(1,0,1,1,0,0,1,1,1,1,1)


screamPlus <- matrix(0L,nrow=nsub, ncol=ntrials)

screamPlus[,1] <- 1


# for (n in 1:dim(screamPlus)[1]) {
#   print(n)
#   screamPlus[n,2:12] <- sample(patts,1,replace=T)
# }


for (n in 1:dim(screamPlus)[1]) {
  
  a <- sample(c(1,4),1)
  
  if (a == 1) {
    screamPlus[n,2:12] <- sc1
  } else if (a == 2) {
    screamPlus[n,2:12] <- sc2
  } else if (a == 3){
    screamPlus[n,2:12] <- sc3
  } else {
    screamPlus[n,2:12] <- sc4
  }
}

2.4 Create dataset for barplot comparing output

library(ggplot2)

mod_comp <- data.frame(model=NA,BIC=NA)

2.5 Model 1: single beta no scaling

2.5.1 notes

Because we use the 1-rescaled expectancy data, no need to try and invert to reinforcement paramters here. As a result we need the stan model to simply be:

alphaPlus[p] =  nothingPlus[p]/ntrials;
alphaMinus[p] =  nothingMinus[p]/ntrials;

2.5.2 run Alex Pike’s stan script for non scaled beta model.

here we try to estimate the alpha paramter of the beta distribution per trial per person per stimulus. (i.e. you have two sufficient paramters for each beta dist, the alpha and beta. we want to estimate the alpha - ).

Eventually we will scale these by the actual ‘value’ of the scream for each person per trial.

Using data loaded in from preliminary tests above.

so this is a beta value per person (assuming the underlying process for the plus and minus are the same)

## decide testing rate (min,med,max or off)
testing('min')

## set up run
stanname='beta_noscaling.stan'

stanfile <- file.path(scriptdir, stanname)

flare_data<-list(ntrials=ntrials,nsub=nsub,nothingPlus = No_scream_p, nothingMinus=No_scream_m,ratingsPlus=plus_scaled,ratingsMinus=minus_scaled)

flare_fit <- stan(file = stanfile, data = flare_data, iter=chain_iter, chains = chain_n) #add working dir?
## 
## SAMPLING FOR MODEL 'beta_noscaling' NOW (CHAIN 1).
## Chain 1: 
## Chain 1: Gradient evaluation took 0.001098 seconds
## Chain 1: 1000 transitions using 10 leapfrog steps per transition would take 10.98 seconds.
## Chain 1: Adjust your expectations accordingly!
## Chain 1: 
## Chain 1: 
## Chain 1: Iteration:   1 / 400 [  0%]  (Warmup)
## Chain 1: Iteration:  40 / 400 [ 10%]  (Warmup)
## Chain 1: Iteration:  80 / 400 [ 20%]  (Warmup)
## Chain 1: Iteration: 120 / 400 [ 30%]  (Warmup)
## Chain 1: Iteration: 160 / 400 [ 40%]  (Warmup)
## Chain 1: Iteration: 200 / 400 [ 50%]  (Warmup)
## Chain 1: Iteration: 201 / 400 [ 50%]  (Sampling)
## Chain 1: Iteration: 240 / 400 [ 60%]  (Sampling)
## Chain 1: Iteration: 280 / 400 [ 70%]  (Sampling)
## Chain 1: Iteration: 320 / 400 [ 80%]  (Sampling)
## Chain 1: Iteration: 360 / 400 [ 90%]  (Sampling)
## Chain 1: Iteration: 400 / 400 [100%]  (Sampling)
## Chain 1: 
## Chain 1:  Elapsed Time: 11.457 seconds (Warm-up)
## Chain 1:                3.57821 seconds (Sampling)
## Chain 1:                15.0352 seconds (Total)
## Chain 1:
save(flare_fit, file=file.path(datadir,'flare_fit_test'))

traceplot(flare_fit,'lp__')

# extract fit data
summary_flare<- summary(flare_fit)
## Warning in sqrt(ess): NaNs produced
# extract model summary data

#flare_loglike<- extract_log_lik(flare_fit, parameter_name = "loglik", merge_chains = TRUE)
## get some basic output descriptions printed to screen

out_describe(summary_flare)
## 
## Attaching package: 'psych'
## The following objects are masked from 'package:scales':
## 
##     alpha, rescale
## The following object is masked from 'package:rstan':
## 
##     lookup
## The following objects are masked from 'package:ggplot2':
## 
##     %+%, alpha
## The following object is masked from 'package:boot':
## 
##     logit
## [1] "400 iterations  on 1 chains. "
## [1] "Estimated 1 Free paramaters per person"
## [1] " "
## [1] "Average Rhat"
## [1] 1
## [1] "beta[1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.67 0.26   0.61    0.63 0.13 0.36 2.58  2.23 3.34     16.2
##      se
## X1 0.01
## [1] " "

2.6 Model 2: single beta scaled

2.6.1 notes

Simple alteration of the first model. We estimate a scaling parameter per person over all trials and apply this to alpha component per participant.

2.6.2 run Alex Pike’s stan script for scaled beta model.

here we try to estimate the alpha paramter of the beta distribution per trial per person per stimulus. (i.e. you have two sufficient paramters for each beta dist, the alpha and beta. we want to estimate the alpha - ).

Eventually we will scale these by the actual ‘value’ of the scream for each person per trial.

Using data loaded in from preliminary tests above.

so this is a beta value per person (assuming the underlying process for the plus and minus are the same)

## decide testing rate (min,med,max or off)
testing('min')

## set up run
stanname='beta_scaling.stan'

stanfile <- file.path(scriptdir, stanname)

flare_data<-list(ntrials=ntrials,nsub=nsub,nothingPlus = No_scream_p, nothingMinus=No_scream_m,ratingsPlus=plus_scaled,ratingsMinus=minus_scaled)

flare_fit <- stan(file = stanfile, data = flare_data, iter=chain_iter, chains = chain_n) #add working dir?
## 
## SAMPLING FOR MODEL 'beta_scaling' NOW (CHAIN 1).
## Chain 1: 
## Chain 1: Gradient evaluation took 0.001307 seconds
## Chain 1: 1000 transitions using 10 leapfrog steps per transition would take 13.07 seconds.
## Chain 1: Adjust your expectations accordingly!
## Chain 1: 
## Chain 1: 
## Chain 1: Iteration:   1 / 400 [  0%]  (Warmup)
## Chain 1: Iteration:  40 / 400 [ 10%]  (Warmup)
## Chain 1: Iteration:  80 / 400 [ 20%]  (Warmup)
## Chain 1: Iteration: 120 / 400 [ 30%]  (Warmup)
## Chain 1: Iteration: 160 / 400 [ 40%]  (Warmup)
## Chain 1: Iteration: 200 / 400 [ 50%]  (Warmup)
## Chain 1: Iteration: 201 / 400 [ 50%]  (Sampling)
## Chain 1: Iteration: 240 / 400 [ 60%]  (Sampling)
## Chain 1: Iteration: 280 / 400 [ 70%]  (Sampling)
## Chain 1: Iteration: 320 / 400 [ 80%]  (Sampling)
## Chain 1: Iteration: 360 / 400 [ 90%]  (Sampling)
## Chain 1: Iteration: 400 / 400 [100%]  (Sampling)
## Chain 1: 
## Chain 1:  Elapsed Time: 23.6451 seconds (Warm-up)
## Chain 1:                53.0175 seconds (Sampling)
## Chain 1:                76.6626 seconds (Total)
## Chain 1:
## Warning: There were 1 transitions after warmup that exceeded the maximum treedepth. Increase max_treedepth above 10. See
## http://mc-stan.org/misc/warnings.html#maximum-treedepth-exceeded
## Warning: Examine the pairs() plot to diagnose sampling problems
save(flare_fit, file=file.path(datadir,'flare_fit_test'))

traceplot(flare_fit,'lp__')

# extract fit data
summary_flare<- summary(flare_fit)

# extract model summary data

#flare_loglike<- extract_log_lik(flare_fit, parameter_name = "loglik", merge_chains = TRUE)
## get some basic output descriptions printed to screen

out_describe(summary_flare)
## [1] "400 iterations  on 1 chains. "
## [1] "Estimated 2 Free paramaters per person"
## [1] " "
## [1] "Average Rhat"
## [1] 1
## [1] "beta[1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.61 1.7   1.16     1.3 0.44 0.37 22.43 22.05 7.12    71.51
##      se
## X1 0.09
## [1] " "
## [1] "scaling[1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 3.76 2.83   3.25    3.41 1.09 0.71 44.87 44.15 9.39   129.04
##      se
## X1 0.15
## [1] " "

2.6.3 run Alex Pike’s stan script for scaled beta model.

here we try to estimate the alpha paramter of the beta distribution per trial per person per stimulus. (i.e. you have two sufficient paramters for each beta dist, the alpha and beta. we want to estimate the alpha - ).

Eventually we will scale these by the actual ‘value’ of the scream for each person per trial.

Using data loaded in from preliminary tests above.

so this is a beta value per person (assuming the underlying process for the plus and minus are the same)

## decide testing rate (min,med,max or off)
testing('min')

## set up run
stanname='beta_withRL.stan'

stanfile <- file.path(scriptdir, stanname)

flare_data<-list(ntrials=ntrials,nsub=nsub,screamPlus = t(screamPlus), screamMinus= t(screamMinus),ratingsPlus=t(plus_scaled),ratingsMinus=t(minus_scaled))

flare_fit <- stan(file = stanfile, data = flare_data, iter=chain_iter, chains = chain_n) #add working dir?
## 
## SAMPLING FOR MODEL 'beta_withRL' NOW (CHAIN 1).
## Chain 1: 
## Chain 1: Gradient evaluation took 0.004328 seconds
## Chain 1: 1000 transitions using 10 leapfrog steps per transition would take 43.28 seconds.
## Chain 1: Adjust your expectations accordingly!
## Chain 1: 
## Chain 1: 
## Chain 1: Iteration:   1 / 400 [  0%]  (Warmup)
## Chain 1: Iteration:  40 / 400 [ 10%]  (Warmup)
## Chain 1: Iteration:  80 / 400 [ 20%]  (Warmup)
## Chain 1: Iteration: 120 / 400 [ 30%]  (Warmup)
## Chain 1: Iteration: 160 / 400 [ 40%]  (Warmup)
## Chain 1: Iteration: 200 / 400 [ 50%]  (Warmup)
## Chain 1: Iteration: 201 / 400 [ 50%]  (Sampling)
## Chain 1: Iteration: 240 / 400 [ 60%]  (Sampling)
## Chain 1: Iteration: 280 / 400 [ 70%]  (Sampling)
## Chain 1: Iteration: 320 / 400 [ 80%]  (Sampling)
## Chain 1: Iteration: 360 / 400 [ 90%]  (Sampling)
## Chain 1: Iteration: 400 / 400 [100%]  (Sampling)
## Chain 1: 
## Chain 1:  Elapsed Time: 38.9101 seconds (Warm-up)
## Chain 1:                52.9128 seconds (Sampling)
## Chain 1:                91.8229 seconds (Total)
## Chain 1:
save(flare_fit, file=file.path(datadir,'flare_fit_test'))

traceplot(flare_fit,'lp__')

# extract fit data
summary_flare <- summary(flare_fit)

# extract model summary data

#flare_loglike<- extract_log_lik(flare_fit, parameter_name = "loglik", merge_chains = TRUE)
## get some basic output descriptions printed to screen

out_describe(summary_flare)
## [1] "400 iterations  on 1 chains. "
## [1] "Estimated 2 Free paramaters per person"
## [1] " "
## [1] "Average Rhat"
## [1] 1
## [1] "alpha[1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.12 0.06    0.1     0.1 0.03 0.05 0.32  0.28 1.71     2.55  0
## [1] " "
## [1] "beta[1]"
## [1] " "
##    vars   n mean sd median trimmed mad min max range  skew kurtosis se
## X1    1 342    0  0      0       0   0   0   0     0 -0.03    -0.42  0
## [1] " "

2.7 Model 3: RL, mean defined, single beta

2.7.1 notes

this model includes an alpha learning paramater per person estimating their learning rate and updating based on it. This model needs a dataset that indicates whether a scream occurred for each trial instead of the proportion of times no scream occurred.

2.7.2 Mean to define shape

this model includes an alpha learning paramater per person estimating their learning rate and updating based on it. This model needs a dataset that indicates whether a scream occurred for each trial instead of the proportion of times no scream occurred.

Alex used this stack post to help solve the shape paramters using mean and sd where we assume that v serves as the mean and beta as the sd.

the equations work out to this:

for shape 1:

\[\alpha = \left(\frac{1-\mu}{\sigma^2} - \frac{1}{\mu}\right)\mu^2\]

for shape 2:

\[\beta=\alpha \left(\frac{1}{\mu}-1\right)\]

Hashed this out as it doesnt run!

# ## decide testing rate (min,med,max or off)
# testing('min')
# 
# ## set up run
# stanname='beta_meansd_RL.stan'
# 
# stanfile <- file.path(scriptdir, stanname)
# 
# flare_data<-list(ntrials=ntrials,nsub=nsub,screamPlus = t(screamPlus), screamMinus=t(screamMinus),ratingsPlus=t(plus_scaled),ratingsMinus=t(minus_scaled))
# 
# flare_fit <- stan(file = stanfile, data = flare_data, iter=chain_iter, chains = chain_n) #add working dir?
# 
# save(flare_fit, file=file.path(datadir,'flare_fit_test'))
# 
# traceplot(flare_fit,'lp__')
# 
# # extract fit data
# summary_flare<- summary(flare_fit)
# 
# # extract model summary data
# 
# #flare_loglike<- extract_log_lik(flare_fit, parameter_name = "loglik", merge_chains = TRUE)
## get some basic output descriptions printed to screen

# out_describe(summary_flare)

On 500 iterations (i.e. test) the variance in alpha is good, but the traceplot is terrible. Model coverges very poorly. We also have to constrain the beta to be betwqeen 0 and 0.0001. Not sure why this is.

when running for 2000 iterations (1000 warmup)…

This results in the following warning;

There were 2644 divergent transitions after warmup. Increasing adapt_delta above 0.8 may help. See http://mc-stan.org/misc/warnings.html#divergent-transitions-after-warmupThere were 4 transitions after warmup that exceeded the maximum treedepth. Increase max_treedepth above 10. See http://mc-stan.org/misc/warnings.html#maximum-treedepth-exceededThere were 4 chains where the estimated Bayesian Fraction of Missing Information was low. See http://mc-stan.org/misc/warnings.html#bfmi-lowExamine the pairs() plot to diagnose sampling problems

2.7.3 Mean definition 2

The above mean definition does not map the data well (terrible traceplot!). I found this from the MRC BSU and have tried defining the beta parameters assuming V == mean in a slighty different way:

for paramater a:

\[\alpha = \mu\beta/(1-\mu)\]

for parameter b:

\[\beta = \mu(1-\mu)^2/\sigma+\mu-1\]

Still using a single beta here.

hashed this out as it doesnt run (saves time)

# ## decide testing rate (min,med,max or off)
# testing('min')
# 
# ## set up run
# stanname='beta_meansd_RL_2.stan'
# 
# stanfile <- file.path(scriptdir, stanname)
# 
# flare_data<-list(ntrials=ntrials,nsub=nsub,screamPlus = t(screamPlus), screamMinus=t(screamMinus),ratingsPlus=t(plus_scaled),ratingsMinus=t(minus_scaled))
# 
# flare_fit <- stan(file = stanfile, data = flare_data, iter=chain_iter, chains = chain_n) #add working dir?
# 
# save(flare_fit, file=file.path(datadir,'flare_fit_test'))
# 
# traceplot(flare_fit,'lp__')
# 
# # extract fit data
# summary_flare<- summary(flare_fit)
# 
# # extract model summary data

#flare_loglike<- extract_log_lik(flare_fit, parameter_name = "loglik", merge_chains = TRUE)
## get some basic output descriptions printed to screen

# out_describe(summary_flare)

2.7.4 Mean definition 3

noted that the shape parameters have slight variations in definition according to discussion here. Updated the script slightly to reflect this based on the reply from ocram.

the first sd term in shape a is changed to variance, so it changes from:

\[\alpha = \left(\frac{1-\mu}{\sigma^2} - \frac{1}{\mu}\right)\mu^2\]

to

\[\alpha = \left(\frac{1-\mu}{\sigma} - \frac{1}{\mu}\right)\mu^2\]

Changes the shape 2 paramter definition from:

\[\beta=\alpha \left(\frac{1}{\mu}-1\right)\]

to

\[\beta = \left(\frac{1-\mu}{\sigma} - \frac{1}{\mu}\right)\mu\left(1-\mu\right)\]

Because this works best, will add loglikelihhod calculation here. Basing this on the probability density function for the beta distribution given the participants actual ratings and sufficient paramters of the distribution per trial.

loglik[p] = loglik[p] + beta_lpdf(ratingsPlus[t,p]|shape1_Plus[t,p],shape2_Plus[t,p]) + beta_lpdf(ratingsMinus[t,p]|shape1_Minus[t,p],shape2_Minus[t,p])

Hashed this out as it doesnt run (saves time)

## decide testing rate (min,med,max or off)
# testing('min')
# 
# ## set up run
# stanname='beta_meansd_RL_3.stan'
# 
# stanfile <- file.path(scriptdir, stanname)
# 
# flare_data<-list(ntrials=ntrials,nsub=nsub,screamPlus = t(screamPlus), screamMinus=t(screamMinus),ratingsPlus=t(plus_scaled),ratingsMinus=t(minus_scaled))
# 
# flare_fit <- stan(file = stanfile, data = flare_data, iter=chain_iter, chains = chain_n) #add working dir?
# 
# save(flare_fit, file=file.path(datadir,'flare_fit_test'))
# 
# traceplot(flare_fit,'lp__')
# 
# # extract fit data
# summary_flare<- summary(flare_fit)
# 
# # extract model summary data
# 
# flare_loglike<- extract_log_lik(flare_fit, parameter_name = "loglik", merge_chains = TRUE)
## get some basic output descriptions printed to screen

# out_describe(summary_flare)

This model is substnatially better than either of the other two. Traceplot suggests that the iterations converge as we would like. However, we still need to massively constrain the beta (i.e. confidence / uncertainty) estimates for it to run, otherwise the starting values drop below zero.

2.7.4.1 Create BIC from log likelihood

## extract log likelihood
# 
# flare_loglike <- extract_log_lik(flare_fit, parameter_name = "loglik", merge_chains = TRUE)
# 
# #calculate BIC
# 
# FLARe_bic<-bic(ntrials,colMeans(flare_loglike),2) #number of parameters in that model e.g. 4)
# 
# ## mean BIC as model comparisons tool:
# 
# print("Mean Bayesian information criterion for model")
# mean(FLARe_bic)

2.7.4.2 Add to bar plot

# 
# mod_comp <- rbind(na.omit(mod_comp,c("Mean 1 beta",mean(FLARe_bic))))
# 

# plot <- ggplot(na.omit(mod_comp),aes(x=model,y=BIC)) +
#   geom_bar(stat = "identity") +
#   coord_flip()
# 
# show(plot)

2.7.5 Mean definition 4

Here I try to define the parameter using simplified mean and precision estimates as per this tutorial. See in particular the paramter estimation on the cubs data.

This results in a relatively simplified paramter estimation compared to model 3.

\[\alpha = \mu * ((\mu * (1-\mu)) / \sigma - 1)\]

where mu is the mean (or value) and sigma is the variance / uncertainty paramter we currently call beta.

and the b (or shape 2) parameter for the distribution is:

\[\beta = (1- \mu) * ((\mu * (1-\mu)) / \sigma - 1)\]

## decide testing rate (min,med,max or off)
testing('min')

## set up run
stanname='beta_meansd_RL_4.stan'

stanfile <- file.path(scriptdir, stanname)

flare_data<-list(ntrials=ntrials,nsub=nsub,screamPlus = t(screamPlus), screamMinus=t(screamMinus),ratingsPlus=t(plus_scaled),ratingsMinus=t(minus_scaled),cdf_scale=cdf_scale)

flare_fit <- stan(file = stanfile, data = flare_data, iter=chain_iter, chains = chain_n) #add working dir?
## recompiling to avoid crashing R session
## 
## SAMPLING FOR MODEL 'beta_meansd_RL_4' NOW (CHAIN 1).
## Chain 1: 
## Chain 1: Gradient evaluation took 0.014984 seconds
## Chain 1: 1000 transitions using 10 leapfrog steps per transition would take 149.84 seconds.
## Chain 1: Adjust your expectations accordingly!
## Chain 1: 
## Chain 1: 
## Chain 1: Iteration:   1 / 400 [  0%]  (Warmup)
## Chain 1: Iteration:  40 / 400 [ 10%]  (Warmup)
## Chain 1: Iteration:  80 / 400 [ 20%]  (Warmup)
## Chain 1: Iteration: 120 / 400 [ 30%]  (Warmup)
## Chain 1: Iteration: 160 / 400 [ 40%]  (Warmup)
## Chain 1: Iteration: 200 / 400 [ 50%]  (Warmup)
## Chain 1: Iteration: 201 / 400 [ 50%]  (Sampling)
## Chain 1: Iteration: 240 / 400 [ 60%]  (Sampling)
## Chain 1: Iteration: 280 / 400 [ 70%]  (Sampling)
## Chain 1: Iteration: 320 / 400 [ 80%]  (Sampling)
## Chain 1: Iteration: 360 / 400 [ 90%]  (Sampling)
## Chain 1: Iteration: 400 / 400 [100%]  (Sampling)
## Chain 1: 
## Chain 1:  Elapsed Time: 74.0258 seconds (Warm-up)
## Chain 1:                37.8058 seconds (Sampling)
## Chain 1:                111.832 seconds (Total)
## Chain 1:
save(flare_fit, file=file.path(datadir,'flare_fit_simpleMean'))

traceplot(flare_fit,'lp__')

# extract fit data
summary_flare <- summary(flare_fit)
## Warning in sqrt(ess): NaNs produced
## get some basic output descriptions printed to screen

out_describe(summary_flare)
## [1] "400 iterations  on 1 chains. "
## [1] "Estimated 97 Free paramaters per person"
## [1] " "
## [1] "Average Rhat"
## [1] 0.9587641
## [1] "alpha[1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.06 0.02   0.06    0.06 0.02 0.01 0.18  0.16 0.62     2.06  0
## [1] " "
## [1] "beta[1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.19 0.05    0.2     0.2 0.06 0.04 0.31  0.27 -0.37    -0.48
##    se
## X1  0
## [1] " "
## [1] "loglik[1]"
## [1] " "
##    vars   n   mean   sd median trimmed  mad    min    max range skew
## X1    1 342 -47.76 5.21 -48.95   -48.3 5.47 -54.34 -33.42 20.92 0.73
##    kurtosis   se
## X1    -0.36 0.28
## [1] " "
## [1] "shape1_Plus[1,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342 0.77 0.35   0.67    0.71 0.18 0.44 3.9  3.46 3.62    21.71
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.81 0.36   0.71    0.75 0.19 0.46 4.01  3.54 3.56    21.27
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.84 0.37   0.73    0.77 0.2 0.48 4.09  3.61 3.53    21.09
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.82 0.36   0.73    0.76 0.2 0.48 3.99  3.51 3.46    19.78
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.83 0.37   0.72    0.76 0.2 0.47 4.08  3.61 3.55    21.26
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.85 0.37   0.74    0.78 0.2 0.49 4.16  3.67 3.54    21.17
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.82 0.36   0.72    0.75 0.2 0.47 4.06   3.6 3.57    21.44
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.77 0.36   0.66    0.71 0.18 0.44 3.95  3.52 3.63    21.91
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.82 0.36   0.71    0.75 0.2 0.46 4.05  3.59 3.57    21.47
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.84 0.37   0.74    0.78 0.2 0.48 4.13  3.65 3.54    21.21
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min max range skew kurtosis   se
## X1    1 342 0.86 0.38   0.75    0.79 0.2 0.49 4.2  3.71 3.54    21.19 0.02
## [1] " "
## [1] "shape1_Plus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad min  max range skew kurtosis   se
## X1    1 342 0.86 0.38   0.75    0.79 0.2 0.5 4.25  3.75 3.55    21.33 0.02
## [1] " "
## [1] "shape1_Minus[1,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342 0.77 0.35   0.67    0.71 0.18 0.44 3.9  3.46 3.62    21.71
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.81 0.36   0.71    0.75 0.19 0.46 4.01  3.54 3.56    21.27
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.84 0.37   0.73    0.77 0.2 0.48 4.09  3.61 3.53    21.09
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.82 0.36   0.73    0.76 0.2 0.48 3.99  3.51 3.46    19.78
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.83 0.37   0.72    0.76 0.2 0.47 4.08  3.61 3.55    21.26
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.85 0.37   0.74    0.78 0.2 0.49 4.16  3.67 3.54    21.17
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.82 0.36   0.72    0.75 0.2 0.47 4.06   3.6 3.57    21.44
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.77 0.36   0.66    0.71 0.18 0.44 3.95  3.52 3.63    21.91
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.82 0.36   0.71    0.75 0.2 0.46 4.05  3.59 3.57    21.47
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.84 0.37   0.74    0.78 0.2 0.48 4.13  3.65 3.54    21.21
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min max range skew kurtosis   se
## X1    1 342 0.86 0.38   0.75    0.79 0.2 0.49 4.2  3.71 3.54    21.19 0.02
## [1] " "
## [1] "shape1_Minus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad min  max range skew kurtosis   se
## X1    1 342 0.86 0.38   0.75    0.79 0.2 0.5 4.25  3.75 3.55    21.33 0.02
## [1] " "
## [1] "shape2_Plus[1,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342 0.77 0.35   0.67    0.71 0.18 0.44 3.9  3.46 3.62    21.71
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.72 0.34   0.62    0.66 0.18 0.41 3.78  3.37 3.68    22.08
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.67 0.34   0.57    0.61 0.17 0.36 3.66  3.29 3.71    22.15
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342 0.68 0.35   0.58    0.63 0.19 0.29 3.8  3.51 3.69    23.05
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.69 0.34   0.59    0.63 0.17 0.39 3.68  3.29  3.7    22.04
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.64 0.33   0.54    0.58 0.16 0.34 3.56  3.22 3.72       22
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.71 0.34   0.61    0.65 0.17 0.4 3.71  3.31 3.68    21.95
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.77 0.35   0.67    0.71 0.18 0.44 3.84  3.41  3.6    21.44
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.72 0.34   0.62    0.66 0.17 0.41 3.72  3.32 3.67    21.93
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342 0.67 0.33   0.56    0.61 0.17 0.37 3.6  3.23 3.72     22.1
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.62 0.33   0.52    0.56 0.16 0.32 3.48  3.16 3.73    21.98
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.57 0.32   0.48    0.52 0.16 0.26 3.36  3.11 3.73    21.68
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[1,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342 0.77 0.35   0.67    0.71 0.18 0.44 3.9  3.46 3.62    21.71
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.72 0.34   0.62    0.66 0.18 0.41 3.78  3.37 3.68    22.08
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.67 0.34   0.57    0.61 0.17 0.36 3.66  3.29 3.71    22.15
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342 0.68 0.35   0.58    0.63 0.19 0.29 3.8  3.51 3.69    23.05
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.69 0.34   0.59    0.63 0.17 0.39 3.68  3.29  3.7    22.04
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.64 0.33   0.54    0.58 0.16 0.34 3.56  3.22 3.72       22
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.71 0.34   0.61    0.65 0.17 0.4 3.71  3.31 3.68    21.95
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.77 0.35   0.67    0.71 0.18 0.44 3.84  3.41  3.6    21.44
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.72 0.34   0.62    0.66 0.17 0.41 3.72  3.32 3.67    21.93
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342 0.67 0.33   0.56    0.61 0.17 0.37 3.6  3.23 3.72     22.1
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.62 0.33   0.52    0.56 0.16 0.32 3.48  3.16 3.73    21.98
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.57 0.32   0.48    0.52 0.16 0.26 3.36  3.11 3.73    21.68
##      se
## X1 0.02
## [1] " "
## [1] "VPlus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad min max range skew kurtosis se
## X1    1 342  0.5  0    0.5     0.5   0 0.5 0.5     0  NaN      NaN  0
## [1] " "
## [1] "VPlus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.53 0.01   0.53    0.53 0.01 0.51 0.59  0.08 0.62     2.06  0
## [1] " "
## [1] "VPlus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.56 0.02   0.56    0.56 0.02 0.51 0.66  0.15 0.51     1.74  0
## [1] " "
## [1] "VPlus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.55 0.04   0.53    0.55 0.02 0.51 0.69  0.18 0.85    -0.15  0
## [1] " "
## [1] "VPlus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.55 0.02   0.55    0.55 0.01 0.51 0.62  0.11 0.35     1.47  0
## [1] " "
## [1] "VPlus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.58 0.02   0.58    0.58 0.02 0.51 0.68  0.17  0.3     1.35  0
## [1] " "
## [1] "VPlus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.54 0.01   0.54    0.54 0.01 0.51 0.56  0.05 -0.37     0.54
##    se
## X1  0
## [1] " "
## [1] "VPlus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range  skew kurtosis se
## X1    1 342  0.5 0.01    0.5     0.5   0 0.46 0.51  0.05 -2.79     12.3  0
## [1] " "
## [1] "VPlus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.53 0.01   0.53    0.53 0.01 0.51 0.56  0.05 -0.37     0.66
##    se
## X1  0
## [1] " "
## [1] "VPlus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.56 0.02   0.56    0.56 0.02 0.51 0.64  0.12 0.08     1.02  0
## [1] " "
## [1] "VPlus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis se
## X1    1 342 0.59 0.03   0.59    0.59 0.02 0.52 0.7  0.18 0.15      1.1  0
## [1] " "
## [1] "VPlus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.61 0.03   0.61    0.61 0.03 0.52 0.75  0.22 0.14     1.06  0
## [1] " "
## [1] "VMinus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad min max range skew kurtosis se
## X1    1 342  0.5  0    0.5     0.5   0 0.5 0.5     0  NaN      NaN  0
## [1] " "
## [1] "VMinus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.53 0.01   0.53    0.53 0.01 0.51 0.59  0.08 0.62     2.06  0
## [1] " "
## [1] "VMinus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.56 0.02   0.56    0.56 0.02 0.51 0.66  0.15 0.51     1.74  0
## [1] " "
## [1] "VMinus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.55 0.04   0.53    0.55 0.02 0.51 0.69  0.18 0.85    -0.15  0
## [1] " "
## [1] "VMinus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.55 0.02   0.55    0.55 0.01 0.51 0.62  0.11 0.35     1.47  0
## [1] " "
## [1] "VMinus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.58 0.02   0.58    0.58 0.02 0.51 0.68  0.17  0.3     1.35  0
## [1] " "
## [1] "VMinus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.54 0.01   0.54    0.54 0.01 0.51 0.56  0.05 -0.37     0.54
##    se
## X1  0
## [1] " "
## [1] "VMinus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range  skew kurtosis se
## X1    1 342  0.5 0.01    0.5     0.5   0 0.46 0.51  0.05 -2.79     12.3  0
## [1] " "
## [1] "VMinus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.53 0.01   0.53    0.53 0.01 0.51 0.56  0.05 -0.37     0.66
##    se
## X1  0
## [1] " "
## [1] "VMinus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.56 0.02   0.56    0.56 0.02 0.51 0.64  0.12 0.08     1.02  0
## [1] " "
## [1] "VMinus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis se
## X1    1 342 0.59 0.03   0.59    0.59 0.02 0.52 0.7  0.18 0.15      1.1  0
## [1] " "
## [1] "VMinus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.61 0.03   0.61    0.61 0.03 0.52 0.75  0.22 0.14     1.06  0
## [1] " "
## [1] "deltaPlus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad min max range skew kurtosis se
## X1    1 342  0.5  0    0.5     0.5   0 0.5 0.5     0  NaN      NaN  0
## [1] " "
## [1] "deltaPlus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.47 0.01   0.47    0.47 0.01 0.41 0.49  0.08 -0.62     2.06
##    se
## X1  0
## [1] " "
## [1] "deltaPlus[3,1]"
## [1] " "
##    vars   n  mean  sd median trimmed  mad   min  max range skew kurtosis
## X1    1 342 -0.07 0.5  -0.52   -0.08 0.12 -0.66 0.49  1.15 0.06       -2
##      se
## X1 0.03
## [1] " "
## [1] "deltaPlus[4,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min  max range  skew kurtosis
## X1    1 342 -0.04 0.53   0.47   -0.03 0.03 -0.69 0.49  1.18 -0.06       -2
##      se
## X1 0.03
## [1] " "
## [1] "deltaPlus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.45 0.02   0.45    0.45 0.01 0.38 0.49  0.11 -0.35     1.47
##    se
## X1  0
## [1] " "
## [1] "deltaPlus[6,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.58 0.02  -0.58   -0.58 0.02 -0.68 -0.51  0.17 -0.3     1.35
##    se
## X1  0
## [1] " "
## [1] "deltaPlus[7,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.54 0.01  -0.54   -0.54 0.01 -0.56 -0.51  0.05 0.37     0.54
##    se
## X1  0
## [1] " "
## [1] "deltaPlus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis se
## X1    1 342  0.5 0.01    0.5     0.5   0 0.49 0.54  0.05 2.79     12.3  0
## [1] " "
## [1] "deltaPlus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.47 0.01   0.47    0.47 0.01 0.44 0.49  0.05 0.37     0.66  0
## [1] " "
## [1] "deltaPlus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.44 0.02   0.44    0.44 0.02 0.36 0.49  0.12 -0.08     1.02
##    se
## X1  0
## [1] " "
## [1] "deltaPlus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range  skew kurtosis se
## X1    1 342 0.41 0.03   0.41    0.41 0.02 0.3 0.48  0.18 -0.15      1.1  0
## [1] " "
## [1] "deltaMinus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad  min  max range skew kurtosis se
## X1    1 342 -0.5  0   -0.5    -0.5   0 -0.5 -0.5     0  NaN      NaN  0
## [1] " "
## [1] "deltaMinus[2,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range  skew
## X1    1 342 -0.53 0.01  -0.53   -0.53 0.01 -0.59 -0.51  0.08 -0.62
##    kurtosis se
## X1     2.06  0
## [1] " "
## [1] "deltaMinus[3,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range  skew
## X1    1 342 -0.56 0.02  -0.56   -0.56 0.02 -0.66 -0.51  0.15 -0.51
##    kurtosis se
## X1     1.74  0
## [1] " "
## [1] "deltaMinus[4,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range  skew
## X1    1 342 -0.55 0.04  -0.53   -0.55 0.02 -0.69 -0.51  0.18 -0.85
##    kurtosis se
## X1    -0.15  0
## [1] " "
## [1] "deltaMinus[5,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range  skew
## X1    1 342 -0.55 0.02  -0.55   -0.55 0.01 -0.62 -0.51  0.11 -0.35
##    kurtosis se
## X1     1.47  0
## [1] " "
## [1] "deltaMinus[6,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.58 0.02  -0.58   -0.58 0.02 -0.68 -0.51  0.17 -0.3     1.35
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[7,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.54 0.01  -0.54   -0.54 0.01 -0.56 -0.51  0.05 0.37     0.54
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad   min   max range skew kurtosis
## X1    1 342 -0.5 0.01   -0.5    -0.5   0 -0.51 -0.46  0.05 2.79     12.3
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[9,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.53 0.01  -0.53   -0.53 0.01 -0.56 -0.51  0.05 0.37     0.66
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[10,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range  skew
## X1    1 342 -0.56 0.02  -0.56   -0.56 0.02 -0.64 -0.51  0.12 -0.08
##    kurtosis se
## X1     1.02  0
## [1] " "
## [1] "deltaMinus[11,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad  min   max range  skew kurtosis
## X1    1 342 -0.59 0.03  -0.59   -0.59 0.02 -0.7 -0.52  0.18 -0.15      1.1
##    se
## X1  0
## [1] " "

2.7.5.1 Create BIC from log likelihood

## extract log likelihood

flare_loglike <- extract_log_lik(flare_fit, parameter_name = "loglik", merge_chains = TRUE)

#calculate BIC

FLARe_bic<-bic(ntrials,colMeans(flare_loglike),2) #number of parameters in that model e.g. 4)

## mean BIC as model comparisons tool:

print("Mean Bayesian information criterion for model")
## [1] "Mean Bayesian information criterion for model"
mean(FLARe_bic)
## [1] -90.55766

2.7.5.2 Add to bar plot

mod_comp <- rbind(mod_comp,c("Means 1 beta",mean(FLARe_bic)))


plot <- ggplot(na.omit(mod_comp),aes(x=model,y=BIC)) +
  geom_bar(stat = "identity") +
  coord_flip()

show(plot)

2.8 Model 4: RL, mode defined, single beta

2.8.1 notes

Used this post to guide this. particularly:

For a beta distribution with shape parameters a and b, the mode is (a-1)/(a+b-2). Suppose we have a desired mode, and we want to determine the corresponding shape parameters. Here’s the solution. First, we express the “certainty” of the estimate in terms of the equivalent prior sample size, k=a+b, with k≥2. The certainty must be at least 2 because it essentially assumes that the prior contains at least one “head” and one “tail,” which is to say that we know each outcome is at least possible. Then a little algebra reveals: a = mode * (k-2) + 1 b = (1-mode) * (k-2) + 1

2.8.2 shape 1 as mode with v and beta as beta shape parameters

For this version we try and estimate the ‘mode’ to be shape 1. KIRSTIN:: explain here

## decide testing rate (min,med,max or off)
# 
# testing('skip')
# 
# ## set up run
# 
# stanname='beta_mode_RL.stan'
# 
# stanfile <- file.path(scriptdir, stanname)
# 
# flare_data<-list(ntrials=ntrials,nsub=nsub,screamPlus = t(screamPlus), screamMinus=t(screamMinus),ratingsPlus=t(plus_scaled),ratingsMinus=t(minus_scaled))
# 
# flare_fit <- stan(file = stanfile, data = flare_data, iter=chain_iter, chains = chain_n) #add working dir?
# 
# save(flare_fit, file=file.path(datadir,'flare_fit_simpleMode'))
# 
# traceplot(flare_fit,'lp__')
# 
# # extract fit data
# summary_flare<- summary(flare_fit)
# 
# # extract model summary data
# 
# #flare_loglike<- extract_log_lik(flare_fit, parameter_name = "loglik", merge_chains = TRUE)
## get some basic output descriptions printed to screen

# out_describe(summary_flare)

2.8.3 v as mode

For this version we assume that V is the mode (above we assumed it serves as the mean) and beta is the certainty aspect (i.e. k)

What this does is basically treat the expected rating (value) as the a parameter for the distribution (scaled by their certainity - beta) and 1-that value as the b parameter (again, scaled by the uncertainty).

so you have a ratio of their selected value per trial (mode across iterations?) to how far from the highest possible choice they are.

## decide testing rate (min,med,max or off)
testing('min')

## set up run
stanname='beta_mode_RL_2.stan'

stanfile <- file.path(scriptdir, stanname)

flare_data<-list(ntrials=ntrials,nsub=nsub,screamPlus = t(screamPlus), screamMinus=t(screamMinus),ratingsPlus=t(plus_scaled),ratingsMinus=t(minus_scaled),cdf_scale=cdf_scale)

flare_fit <- stan(file = stanfile, data = flare_data, iter=chain_iter, chains = chain_n) #add working dir?
## 
## SAMPLING FOR MODEL 'beta_mode_RL_2' NOW (CHAIN 1).
## Chain 1: 
## Chain 1: Gradient evaluation took 0.009335 seconds
## Chain 1: 1000 transitions using 10 leapfrog steps per transition would take 93.35 seconds.
## Chain 1: Adjust your expectations accordingly!
## Chain 1: 
## Chain 1: 
## Chain 1: Iteration:   1 / 400 [  0%]  (Warmup)
## Chain 1: Iteration:  40 / 400 [ 10%]  (Warmup)
## Chain 1: Iteration:  80 / 400 [ 20%]  (Warmup)
## Chain 1: Iteration: 120 / 400 [ 30%]  (Warmup)
## Chain 1: Iteration: 160 / 400 [ 40%]  (Warmup)
## Chain 1: Iteration: 200 / 400 [ 50%]  (Warmup)
## Chain 1: Iteration: 201 / 400 [ 50%]  (Sampling)
## Chain 1: Iteration: 240 / 400 [ 60%]  (Sampling)
## Chain 1: Iteration: 280 / 400 [ 70%]  (Sampling)
## Chain 1: Iteration: 320 / 400 [ 80%]  (Sampling)
## Chain 1: Iteration: 360 / 400 [ 90%]  (Sampling)
## Chain 1: Iteration: 400 / 400 [100%]  (Sampling)
## Chain 1: 
## Chain 1:  Elapsed Time: 50.1484 seconds (Warm-up)
## Chain 1:                32.2323 seconds (Sampling)
## Chain 1:                82.3808 seconds (Total)
## Chain 1:
save(flare_fit, file=file.path(datadir,'flare_fit_test'))

traceplot(flare_fit,'lp__')

# extract fit data
summary_flare<- summary(flare_fit)
## get some basic output descriptions printed to screen

out_describe(summary_flare)
## [1] "400 iterations  on 1 chains. "
## [1] "Estimated 97 Free paramaters per person"
## [1] " "
## [1] "Average Rhat"
## [1] 0.9587641
## [1] "alpha[1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.07 0.03   0.06    0.06 0.02 0.01 0.21  0.21 1.07     3.33  0
## [1] " "
## [1] "beta[1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 1.68 0.81   1.44    1.54 0.41 0.94 8.86  7.91 3.68     22.1
##      se
## X1 0.04
## [1] " "
## [1] "loglik[1]"
## [1] " "
##    vars   n   mean   sd median trimmed mad    min    max range skew
## X1    1 342 -48.43 4.81  -49.6  -48.96   5 -54.18 -34.98  19.2 0.79
##    kurtosis   se
## X1    -0.26 0.26
## [1] " "
## [1] "shape1_Plus[1,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.84 0.4   0.72    0.77 0.21 0.47 4.43  3.96 3.68     22.1
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.89 0.41   0.77    0.82 0.22 0.5 4.56  4.05 3.64    21.94
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.94 0.42   0.82    0.86 0.23 0.53 4.68  4.15  3.6    21.66
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.92 0.42   0.81    0.85 0.23 0.52 4.54  4.02 3.45    19.39
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.92 0.42    0.8    0.84 0.23 0.51 4.66  4.15 3.63       22
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.96 0.43   0.84    0.89 0.24 0.54 4.78  4.24 3.59    21.73
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342  0.9 0.42   0.77    0.82 0.22 0.5 4.64  4.13 3.66    22.21
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342 0.84 0.41   0.72    0.77 0.22 0.47 4.5  4.03 3.68    22.25
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.89 0.42   0.77    0.82 0.22 0.5 4.62  4.12 3.66    22.24
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.94 0.42   0.82    0.86 0.23 0.53 4.74  4.21 3.63    22.02
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.98 0.43   0.86     0.9 0.24 0.55 4.86   4.3 3.59    21.73
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 1.02 0.44   0.89    0.94 0.25 0.58 4.97  4.39 3.55    21.47
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[1,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.84 0.4   0.72    0.77 0.21 0.47 4.43  3.96 3.68     22.1
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.89 0.41   0.77    0.82 0.22 0.5 4.56  4.05 3.64    21.94
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.94 0.42   0.82    0.86 0.23 0.53 4.68  4.15  3.6    21.66
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.92 0.42   0.81    0.85 0.23 0.52 4.54  4.02 3.45    19.39
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.92 0.42    0.8    0.84 0.23 0.51 4.66  4.15 3.63       22
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.96 0.43   0.84    0.89 0.24 0.54 4.78  4.24 3.59    21.73
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342  0.9 0.42   0.77    0.82 0.22 0.5 4.64  4.13 3.66    22.21
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342 0.84 0.41   0.72    0.77 0.22 0.47 4.5  4.03 3.68    22.25
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.89 0.42   0.77    0.82 0.22 0.5 4.62  4.12 3.66    22.24
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.94 0.42   0.82    0.86 0.23 0.53 4.74  4.21 3.63    22.02
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.98 0.43   0.86     0.9 0.24 0.55 4.86   4.3 3.59    21.73
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 1.02 0.44   0.89    0.94 0.25 0.58 4.97  4.39 3.55    21.47
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[1,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.84 0.4   0.72    0.77 0.21 0.47 4.43  3.96 3.68     22.1
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[2,1]"
## [1] " "
##    vars   n mean  sd median trimmed mad  min max range skew kurtosis   se
## X1    1 342 0.79 0.4   0.67    0.72 0.2 0.44 4.3  3.86 3.69    21.99 0.02
## [1] " "
## [1] "shape2_Plus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.75 0.39   0.63    0.68 0.19 0.39 4.18  3.79 3.68    21.69
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[4,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.76 0.4   0.64    0.69 0.21 0.33 4.32  3.99 3.72    22.97
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.77 0.39   0.65    0.69 0.19 0.42 4.19  3.78 3.69     21.8
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.72 0.39   0.61    0.65 0.19 0.36 4.08  3.71 3.68    21.45
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.79 0.39   0.67    0.71 0.19 0.44 4.22  3.78 3.69    21.89
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[8,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.84 0.4   0.73    0.77 0.21 0.47 4.36  3.88 3.66    21.83
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.79 0.39   0.67    0.72 0.2 0.44 4.23  3.79 3.69    21.87
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.74 0.39   0.62    0.67 0.19 0.4 4.11  3.71 3.69    21.63
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342  0.7 0.38   0.59    0.63 0.19 0.35   4  3.65 3.67    21.26
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.67 0.38   0.55     0.6 0.18 0.3 3.89  3.59 3.65    20.83
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[1,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.84 0.4   0.72    0.77 0.21 0.47 4.43  3.96 3.68     22.1
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[2,1]"
## [1] " "
##    vars   n mean  sd median trimmed mad  min max range skew kurtosis   se
## X1    1 342 0.79 0.4   0.67    0.72 0.2 0.44 4.3  3.86 3.69    21.99 0.02
## [1] " "
## [1] "shape2_Minus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.75 0.39   0.63    0.68 0.19 0.39 4.18  3.79 3.68    21.69
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[4,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.76 0.4   0.64    0.69 0.21 0.33 4.32  3.99 3.72    22.97
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.77 0.39   0.65    0.69 0.19 0.42 4.19  3.78 3.69     21.8
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.72 0.39   0.61    0.65 0.19 0.36 4.08  3.71 3.68    21.45
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.79 0.39   0.67    0.71 0.19 0.44 4.22  3.78 3.69    21.89
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[8,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.84 0.4   0.73    0.77 0.21 0.47 4.36  3.88 3.66    21.83
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.79 0.39   0.67    0.72 0.2 0.44 4.23  3.79 3.69    21.87
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.74 0.39   0.62    0.67 0.19 0.4 4.11  3.71 3.69    21.63
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342  0.7 0.38   0.59    0.63 0.19 0.35   4  3.65 3.67    21.26
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.67 0.38   0.55     0.6 0.18 0.3 3.89  3.59 3.65    20.83
##      se
## X1 0.02
## [1] " "
## [1] "VPlus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad min max range skew kurtosis se
## X1    1 342  0.5  0    0.5     0.5   0 0.5 0.5     0  NaN      NaN  0
## [1] " "
## [1] "VPlus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis se
## X1    1 342 0.53 0.01   0.53    0.53 0.01 0.5 0.61   0.1 1.07     3.33  0
## [1] " "
## [1] "VPlus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.56 0.03   0.56    0.56 0.02 0.51 0.69  0.18  0.9     2.63  0
## [1] " "
## [1] "VPlus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.55 0.04   0.53    0.55 0.02 0.51 0.74  0.23 1.18      1.2  0
## [1] " "
## [1] "VPlus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.55 0.02   0.55    0.55 0.02 0.51 0.64  0.13  0.5     1.67  0
## [1] " "
## [1] "VPlus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.58 0.03   0.58    0.58 0.02 0.51 0.71   0.2 0.51      1.5  0
## [1] " "
## [1] "VPlus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.54 0.01   0.54    0.54 0.01 0.51 0.56  0.05 -0.38     0.34
##    se
## X1  0
## [1] " "
## [1] "VPlus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range  skew kurtosis se
## X1    1 342  0.5 0.01    0.5     0.5   0 0.42 0.51  0.09 -3.71    20.66  0
## [1] " "
## [1] "VPlus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.53 0.01   0.53    0.53 0.01 0.51 0.56  0.05 -0.33     0.45
##    se
## X1  0
## [1] " "
## [1] "VPlus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.56 0.02   0.56    0.56 0.02 0.51 0.66  0.14 0.33     1.17  0
## [1] " "
## [1] "VPlus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.59 0.03   0.59    0.59 0.03 0.52 0.73  0.21 0.44     1.35  0
## [1] " "
## [1] "VPlus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.61 0.04   0.61    0.61 0.03 0.52 0.78  0.26 0.41     1.26  0
## [1] " "
## [1] "VMinus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad min max range skew kurtosis se
## X1    1 342  0.5  0    0.5     0.5   0 0.5 0.5     0  NaN      NaN  0
## [1] " "
## [1] "VMinus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis se
## X1    1 342 0.53 0.01   0.53    0.53 0.01 0.5 0.61   0.1 1.07     3.33  0
## [1] " "
## [1] "VMinus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.56 0.03   0.56    0.56 0.02 0.51 0.69  0.18  0.9     2.63  0
## [1] " "
## [1] "VMinus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.55 0.04   0.53    0.55 0.02 0.51 0.74  0.23 1.18      1.2  0
## [1] " "
## [1] "VMinus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.55 0.02   0.55    0.55 0.02 0.51 0.64  0.13  0.5     1.67  0
## [1] " "
## [1] "VMinus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.58 0.03   0.58    0.58 0.02 0.51 0.71   0.2 0.51      1.5  0
## [1] " "
## [1] "VMinus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.54 0.01   0.54    0.54 0.01 0.51 0.56  0.05 -0.38     0.34
##    se
## X1  0
## [1] " "
## [1] "VMinus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range  skew kurtosis se
## X1    1 342  0.5 0.01    0.5     0.5   0 0.42 0.51  0.09 -3.71    20.66  0
## [1] " "
## [1] "VMinus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.53 0.01   0.53    0.53 0.01 0.51 0.56  0.05 -0.33     0.45
##    se
## X1  0
## [1] " "
## [1] "VMinus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.56 0.02   0.56    0.56 0.02 0.51 0.66  0.14 0.33     1.17  0
## [1] " "
## [1] "VMinus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.59 0.03   0.59    0.59 0.03 0.52 0.73  0.21 0.44     1.35  0
## [1] " "
## [1] "VMinus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.61 0.04   0.61    0.61 0.03 0.52 0.78  0.26 0.41     1.26  0
## [1] " "
## [1] "deltaPlus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad min max range skew kurtosis se
## X1    1 342  0.5  0    0.5     0.5   0 0.5 0.5     0  NaN      NaN  0
## [1] " "
## [1] "deltaPlus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range  skew kurtosis se
## X1    1 342 0.47 0.01   0.47    0.47 0.01 0.39 0.5   0.1 -1.07     3.33  0
## [1] " "
## [1] "deltaPlus[3,1]"
## [1] " "
##    vars   n  mean  sd median trimmed  mad   min  max range skew kurtosis
## X1    1 342 -0.08 0.5  -0.52   -0.08 0.14 -0.69 0.49  1.18 0.06    -1.99
##      se
## X1 0.03
## [1] " "
## [1] "deltaPlus[4,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min  max range  skew kurtosis
## X1    1 342 -0.04 0.53   0.47   -0.03 0.04 -0.74 0.49  1.23 -0.06    -1.99
##      se
## X1 0.03
## [1] " "
## [1] "deltaPlus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.45 0.02   0.45    0.45 0.02 0.36 0.49  0.13 -0.5     1.67  0
## [1] " "
## [1] "deltaPlus[6,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range  skew
## X1    1 342 -0.58 0.03  -0.58   -0.58 0.02 -0.71 -0.51   0.2 -0.51
##    kurtosis se
## X1      1.5  0
## [1] " "
## [1] "deltaPlus[7,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.54 0.01  -0.54   -0.54 0.01 -0.56 -0.51  0.05 0.38     0.34
##    se
## X1  0
## [1] " "
## [1] "deltaPlus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis se
## X1    1 342  0.5 0.01    0.5     0.5   0 0.49 0.58  0.09 3.71    20.66  0
## [1] " "
## [1] "deltaPlus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.47 0.01   0.47    0.47 0.01 0.44 0.49  0.05 0.33     0.45  0
## [1] " "
## [1] "deltaPlus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.44 0.02   0.44    0.44 0.02 0.34 0.49  0.14 -0.33     1.17
##    se
## X1  0
## [1] " "
## [1] "deltaPlus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.41 0.03   0.41    0.41 0.03 0.27 0.48  0.21 -0.44     1.35
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad  min  max range skew kurtosis se
## X1    1 342 -0.5  0   -0.5    -0.5   0 -0.5 -0.5     0  NaN      NaN  0
## [1] " "
## [1] "deltaMinus[2,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min  max range  skew kurtosis
## X1    1 342 -0.53 0.01  -0.53   -0.53 0.01 -0.61 -0.5   0.1 -1.07     3.33
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[3,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.56 0.03  -0.56   -0.56 0.02 -0.69 -0.51  0.18 -0.9     2.63
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[4,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range  skew
## X1    1 342 -0.55 0.04  -0.53   -0.55 0.02 -0.74 -0.51  0.23 -1.18
##    kurtosis se
## X1      1.2  0
## [1] " "
## [1] "deltaMinus[5,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.55 0.02  -0.55   -0.55 0.02 -0.64 -0.51  0.13 -0.5     1.67
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[6,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range  skew
## X1    1 342 -0.58 0.03  -0.58   -0.58 0.02 -0.71 -0.51   0.2 -0.51
##    kurtosis se
## X1      1.5  0
## [1] " "
## [1] "deltaMinus[7,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.54 0.01  -0.54   -0.54 0.01 -0.56 -0.51  0.05 0.38     0.34
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad   min   max range skew kurtosis
## X1    1 342 -0.5 0.01   -0.5    -0.5   0 -0.51 -0.42  0.09 3.71    20.66
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[9,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.53 0.01  -0.53   -0.53 0.01 -0.56 -0.51  0.05 0.33     0.45
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[10,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range  skew
## X1    1 342 -0.56 0.02  -0.56   -0.56 0.02 -0.66 -0.51  0.14 -0.33
##    kurtosis se
## X1     1.17  0
## [1] " "
## [1] "deltaMinus[11,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range  skew
## X1    1 342 -0.59 0.03  -0.59   -0.59 0.03 -0.73 -0.52  0.21 -0.44
##    kurtosis se
## X1     1.35  0
## [1] " "

This works, but there is not a lot of variance in the alpha paramter when described by mode mean 0.49; sd = 0.06. Compared to defined by mean where mean is 0.54 and sd is 0.26.

However there is a lot of variation in the beta paramter (mean -7.21, sd = 134.74)

2.8.3.1 Create BIC from log likelihood

## extract log likelihood

flare_loglike <- extract_log_lik(flare_fit, parameter_name = "loglik", merge_chains = TRUE)

#calculate BIC

FLARe_bic<-bic(ntrials,colMeans(flare_loglike),2) #number of parameters in that model e.g. 4)

## mean BIC as model comparisons tool:

print("Mean Bayesian information criterion for model")
## [1] "Mean Bayesian information criterion for model"
mean(FLARe_bic)
## [1] -91.8878

2.8.3.2 Add to bar plot

# 
 mod_comp <- rbind(na.omit(mod_comp),c("Mode 1 beta",mean(FLARe_bic)))
# 

plot <- ggplot(na.omit(mod_comp),aes(x=model,y=BIC)) +
  geom_bar(stat = "identity") +
  coord_flip()
#
show(plot)

## Model 5: RL mean defined,two beta {.tabset}

2.8.4 mean sd RL model with beta per stimulus

RL model adding a beta per stimulus to Alex’s model

## decide testing rate (min,med,max or off)
testing('min')

## set up run
stanname='beta_meansd_2beta_RL.stan'

stanfile <- file.path(scriptdir, stanname)

flare_data<-list(ntrials=ntrials,nsub=nsub,screamPlus = t(screamPlus), screamMinus= t(screamMinus),ratingsPlus=t(plus_scaled),ratingsMinus=t(minus_scaled),cdf_scale=cdf_scale)

flare_fit <- stan(file = stanfile, data = flare_data, iter=chain_iter,warmup = warm_up, chains = chain_n) #add working dir?
## 
## SAMPLING FOR MODEL 'beta_meansd_2beta_RL' NOW (CHAIN 1).
## Chain 1: 
## Chain 1: Gradient evaluation took 0.007267 seconds
## Chain 1: 1000 transitions using 10 leapfrog steps per transition would take 72.67 seconds.
## Chain 1: Adjust your expectations accordingly!
## Chain 1: 
## Chain 1: 
## Chain 1: WARNING: There aren't enough warmup iterations to fit the
## Chain 1:          three stages of adaptation as currently configured.
## Chain 1:          Reducing each adaptation stage to 15%/75%/10% of
## Chain 1:          the given number of warmup iterations:
## Chain 1:            init_buffer = 15
## Chain 1:            adapt_window = 75
## Chain 1:            term_buffer = 10
## Chain 1: 
## Chain 1: Iteration:   1 / 400 [  0%]  (Warmup)
## Chain 1: Iteration:  40 / 400 [ 10%]  (Warmup)
## Chain 1: Iteration:  80 / 400 [ 20%]  (Warmup)
## Chain 1: Iteration: 101 / 400 [ 25%]  (Sampling)
## Chain 1: Iteration: 140 / 400 [ 35%]  (Sampling)
## Chain 1: Iteration: 180 / 400 [ 45%]  (Sampling)
## Chain 1: Iteration: 220 / 400 [ 55%]  (Sampling)
## Chain 1: Iteration: 260 / 400 [ 65%]  (Sampling)
## Chain 1: Iteration: 300 / 400 [ 75%]  (Sampling)
## Chain 1: Iteration: 340 / 400 [ 85%]  (Sampling)
## Chain 1: Iteration: 380 / 400 [ 95%]  (Sampling)
## Chain 1: Iteration: 400 / 400 [100%]  (Sampling)
## Chain 1: 
## Chain 1:  Elapsed Time: 36.7089 seconds (Warm-up)
## Chain 1:                112.388 seconds (Sampling)
## Chain 1:                149.097 seconds (Total)
## Chain 1:
save(flare_fit, file=file.path(datadir,'flare_fit_2betaMean'))

traceplot(flare_fit,'lp__')

# extract fit data
summary_flare <- summary(flare_fit)
## get some basic output descriptions printed to screen

out_describe(summary_flare)
## [1] "400 iterations  on 1 chains. "
## [1] "Estimated 98 Free paramaters per person"
## [1] " "
## [1] "Average Rhat"
## [1] 0.9591849
## [1] "alpha[1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.08 0.02   0.08    0.08 0.01 0.01 0.17  0.16 0.03     2.41  0
## [1] " "
## [1] "beta[1,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range  skew kurtosis se
## X1    1 342 0.22 0.08   0.22    0.22 0.07 0.02 0.4  0.38 -0.23    -0.37  0
## [1] " "
## [1] "beta[172,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.21 0.07   0.22    0.22 0.07 0.02 0.38  0.36 -0.23    -0.38
##    se
## X1  0
## [1] " "
## [1] "loglik[1]"
## [1] " "
##    vars   n   mean   sd median trimmed  mad   min   max range skew
## X1    1 342 -47.14 5.21 -48.29  -47.65 5.51 -54.5 -32.5    22 0.74
##    kurtosis   se
## X1    -0.27 0.28
## [1] " "
## [1] "shape1_Plus[1,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.65 0.35   0.55    0.59 0.15 0.36 2.88  2.52 3.45     15.3
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342 0.69 0.36   0.59    0.62 0.17 0.39 3.1  2.71 3.45    15.59
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.72 0.37   0.61    0.64 0.17 0.4 3.22  2.82 3.46    15.76
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342  0.7 0.36    0.6    0.63 0.17 0.39 3.03  2.65  3.4    14.87
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.71 0.37   0.61    0.64 0.17 0.39 3.19   2.8 3.46    15.76
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.73 0.37   0.62    0.65 0.17 0.41 3.27  2.86 3.47    15.79
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342  0.7 0.36    0.6    0.63 0.16 0.39 3.13  2.74 3.46    15.63
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.65 0.35   0.55    0.58 0.15 0.35 2.87  2.52 3.45    15.28
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342  0.7 0.36   0.59    0.62 0.16 0.39 3.11  2.72 3.46    15.58
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.72 0.37   0.62    0.65 0.17 0.4 3.24  2.84 3.46    15.74
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.73 0.38   0.62    0.66 0.17 0.41 3.29  2.88 3.46    15.76
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Plus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.73 0.38   0.62    0.66 0.17 0.41 3.28  2.87 3.46    15.65
##      se
## X1 0.02
## [1] " "
## [1] "shape1_Minus[1,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 1.05 0.94   0.76    0.85 0.26 0.41 9.06  8.65 4.73    29.18
##      se
## X1 0.05
## [1] " "
## [1] "shape1_Minus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 1.12 0.98   0.82    0.91 0.27 0.44 9.34   8.9 4.58    27.23
##      se
## X1 0.05
## [1] " "
## [1] "shape1_Minus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 1.15 1.01   0.85    0.94 0.28 0.46 9.58  9.12 4.56    26.96
##      se
## X1 0.05
## [1] " "
## [1] "shape1_Minus[4,1]"
## [1] " "
##    vars   n mean sd median trimmed  mad  min  max range skew kurtosis   se
## X1    1 342 1.13  1   0.81    0.92 0.26 0.43 9.78  9.34 4.73    29.31 0.05
## [1] " "
## [1] "shape1_Minus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 1.14 1.01   0.84    0.93 0.28 0.45 9.53  9.08 4.57    27.07
##      se
## X1 0.05
## [1] " "
## [1] "shape1_Minus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 1.17 1.02   0.86    0.95 0.28 0.46 9.74  9.27 4.58     27.2
##      se
## X1 0.06
## [1] " "
## [1] "shape1_Minus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 1.12 0.99   0.82    0.91 0.28 0.44 9.48  9.04 4.63    27.96
##      se
## X1 0.05
## [1] " "
## [1] "shape1_Minus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 1.03 0.94   0.75    0.84 0.26 0.4 9.19  8.79 4.86    30.82
##      se
## X1 0.05
## [1] " "
## [1] "shape1_Minus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 1.12 0.99   0.82    0.91 0.28 0.44 9.46  9.02 4.64    27.99
##      se
## X1 0.05
## [1] " "
## [1] "shape1_Minus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 1.16 1.02   0.86    0.95 0.28 0.46 9.68  9.22 4.57    27.14
##      se
## X1 0.06
## [1] " "
## [1] "shape1_Minus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 1.18 1.03   0.87    0.96 0.28 0.47 9.87   9.4  4.6    27.55
##      se
## X1 0.06
## [1] " "
## [1] "shape1_Minus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.17 1.04   0.86    0.96 0.29 0.46 10.01  9.55 4.69    28.69
##      se
## X1 0.06
## [1] " "
## [1] "shape2_Plus[1,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.65 0.35   0.55    0.59 0.15 0.36 2.88  2.52 3.45     15.3
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342  0.6 0.33    0.5    0.54 0.14 0.33 2.72  2.39 3.44    15.03
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.56 0.32   0.46    0.49 0.13 0.29 2.69   2.4 3.44    14.94
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.57 0.33   0.48    0.51 0.15 0.25 2.71  2.45 3.44    15.68
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.58 0.32   0.49    0.51 0.14 0.3 2.69  2.38 3.43    14.88
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.53 0.31   0.43    0.47 0.12 0.27 2.66  2.39 3.44    14.97
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342  0.6 0.33    0.5    0.54 0.14 0.33 2.69  2.36 3.44    15.02
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.66 0.34   0.56    0.59 0.16 0.37 2.88  2.51 3.44     15.3
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.61 0.33   0.51    0.54 0.14 0.33 2.69  2.36 3.44    15.06
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.56 0.32   0.46    0.49 0.13 0.29 2.66  2.37 3.45    15.01
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[11,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.51 0.3   0.42    0.45 0.12 0.26 2.63  2.38 3.46    15.19
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342 0.47 0.29   0.38    0.41 0.11 0.22 2.6  2.39 3.49    15.59
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[1,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 1.05 0.94   0.76    0.85 0.26 0.41 9.06  8.65 4.73    29.18
##      se
## X1 0.05
## [1] " "
## [1] "shape2_Minus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.95 0.89   0.68    0.77 0.23 0.36 8.75  8.39    5    32.56
##      se
## X1 0.05
## [1] " "
## [1] "shape2_Minus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.86 0.84    0.6    0.69 0.21 0.32 8.43  8.12 5.26    35.83
##      se
## X1 0.05
## [1] " "
## [1] "shape2_Minus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342  0.9 0.86   0.64    0.72 0.24 0.26 8.11  7.85 4.76     29.4
##      se
## X1 0.05
## [1] " "
## [1] "shape2_Minus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.91 0.86   0.64    0.73 0.22 0.33 8.51  8.18 5.09    33.73
##      se
## X1 0.05
## [1] " "
## [1] "shape2_Minus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min max range skew kurtosis   se
## X1    1 342 0.82 0.82   0.57    0.65 0.2 0.29 8.2   7.9 5.35     36.9 0.04
## [1] " "
## [1] "shape2_Minus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.95 0.88   0.68    0.77 0.23 0.36 8.59  8.22 4.88    31.12
##      se
## X1 0.05
## [1] " "
## [1] "shape2_Minus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 1.05 0.94   0.77    0.85 0.26 0.41 8.93  8.51 4.63    27.88
##      se
## X1 0.05
## [1] " "
## [1] "shape2_Minus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.96 0.89   0.69    0.78 0.23 0.37 8.61  8.25 4.87    31.02
##      se
## X1 0.05
## [1] " "
## [1] "shape2_Minus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.87 0.84   0.61     0.7 0.21 0.32 8.29  7.97 5.16    34.59
##      se
## X1 0.05
## [1] " "
## [1] "shape2_Minus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.78 0.79   0.55    0.62 0.19 0.28 7.98   7.7 5.41    37.63
##      se
## X1 0.04
## [1] " "
## [1] "shape2_Minus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.71 0.76   0.49    0.56 0.18 0.23 7.66  7.43  5.6    39.95
##      se
## X1 0.04
## [1] " "
## [1] "VPlus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad min max range skew kurtosis se
## X1    1 342  0.5  0    0.5     0.5   0 0.5 0.5     0  NaN      NaN  0
## [1] " "
## [1] "VPlus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.54 0.01   0.54    0.54 0.01 0.51 0.58  0.08 0.03     2.41  0
## [1] " "
## [1] "VPlus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.57 0.02   0.57    0.57 0.01 0.51 0.65  0.14 -0.1     2.32  0
## [1] " "
## [1] "VPlus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.56 0.04   0.53    0.56 0.02 0.51 0.69  0.18  0.5    -1.16  0
## [1] " "
## [1] "VPlus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.56 0.02   0.56    0.56 0.01 0.51 0.62  0.11 -0.03     1.94
##    se
## X1  0
## [1] " "
## [1] "VPlus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.59 0.02   0.59    0.59 0.02 0.52 0.68  0.16 -0.2      2.2  0
## [1] " "
## [1] "VPlus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.54 0.01   0.54    0.54 0.01 0.51 0.56  0.05 -0.9     2.09  0
## [1] " "
## [1] "VPlus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range  skew kurtosis se
## X1    1 342  0.5 0.01    0.5     0.5   0 0.46 0.51  0.05 -1.68     5.33  0
## [1] " "
## [1] "VPlus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.54 0.01   0.54    0.54 0.01 0.51 0.56  0.05 -0.92     2.46
##    se
## X1  0
## [1] " "
## [1] "VPlus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.57 0.02   0.57    0.57 0.01 0.51 0.63  0.12 -0.51     2.36
##    se
## X1  0
## [1] " "
## [1] "VPlus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342  0.6 0.03    0.6     0.6 0.02 0.52 0.69  0.17 -0.46     2.35
##    se
## X1  0
## [1] " "
## [1] "VPlus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.63 0.03   0.63    0.63 0.02 0.53 0.74  0.21 -0.5     2.36  0
## [1] " "
## [1] "VMinus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad min max range skew kurtosis se
## X1    1 342  0.5  0    0.5     0.5   0 0.5 0.5     0  NaN      NaN  0
## [1] " "
## [1] "VMinus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.54 0.01   0.54    0.54 0.01 0.51 0.58  0.08 0.03     2.41  0
## [1] " "
## [1] "VMinus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.57 0.02   0.57    0.57 0.01 0.51 0.65  0.14 -0.1     2.32  0
## [1] " "
## [1] "VMinus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.56 0.04   0.53    0.56 0.02 0.51 0.69  0.18  0.5    -1.16  0
## [1] " "
## [1] "VMinus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.56 0.02   0.56    0.56 0.01 0.51 0.62  0.11 -0.03     1.94
##    se
## X1  0
## [1] " "
## [1] "VMinus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.59 0.02   0.59    0.59 0.02 0.52 0.68  0.16 -0.2      2.2  0
## [1] " "
## [1] "VMinus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.54 0.01   0.54    0.54 0.01 0.51 0.56  0.05 -0.9     2.09  0
## [1] " "
## [1] "VMinus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range  skew kurtosis se
## X1    1 342  0.5 0.01    0.5     0.5   0 0.46 0.51  0.05 -1.68     5.33  0
## [1] " "
## [1] "VMinus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.54 0.01   0.54    0.54 0.01 0.51 0.56  0.05 -0.92     2.46
##    se
## X1  0
## [1] " "
## [1] "VMinus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.57 0.02   0.57    0.57 0.01 0.51 0.63  0.12 -0.51     2.36
##    se
## X1  0
## [1] " "
## [1] "VMinus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342  0.6 0.03    0.6     0.6 0.02 0.52 0.69  0.17 -0.46     2.35
##    se
## X1  0
## [1] " "
## [1] "VMinus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.63 0.03   0.63    0.63 0.02 0.53 0.74  0.21 -0.5     2.36  0
## [1] " "
## [1] "deltaPlus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad min max range skew kurtosis se
## X1    1 342  0.5  0    0.5     0.5   0 0.5 0.5     0  NaN      NaN  0
## [1] " "
## [1] "deltaPlus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.46 0.01   0.46    0.46 0.01 0.42 0.49  0.08 -0.03     2.41
##    se
## X1  0
## [1] " "
## [1] "deltaPlus[3,1]"
## [1] " "
##    vars   n  mean  sd median trimmed  mad   min  max range skew kurtosis
## X1    1 342 -0.09 0.5  -0.53   -0.09 0.14 -0.65 0.49  1.14 0.06       -2
##      se
## X1 0.03
## [1] " "
## [1] "deltaPlus[4,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min  max range  skew kurtosis
## X1    1 342 -0.05 0.54   0.47   -0.04 0.03 -0.69 0.49  1.18 -0.06       -2
##      se
## X1 0.03
## [1] " "
## [1] "deltaPlus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.44 0.02   0.44    0.44 0.01 0.38 0.49  0.11 0.03     1.94  0
## [1] " "
## [1] "deltaPlus[6,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.59 0.02  -0.59   -0.59 0.02 -0.68 -0.52  0.16  0.2      2.2
##    se
## X1  0
## [1] " "
## [1] "deltaPlus[7,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.54 0.01  -0.54   -0.54 0.01 -0.56 -0.51  0.05  0.9     2.09
##    se
## X1  0
## [1] " "
## [1] "deltaPlus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis se
## X1    1 342  0.5 0.01    0.5     0.5   0 0.49 0.54  0.05 1.68     5.33  0
## [1] " "
## [1] "deltaPlus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.46 0.01   0.46    0.46 0.01 0.44 0.49  0.05 0.92     2.46  0
## [1] " "
## [1] "deltaPlus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.43 0.02   0.43    0.43 0.01 0.37 0.49  0.12 0.51     2.36  0
## [1] " "
## [1] "deltaPlus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342  0.4 0.03    0.4     0.4 0.02 0.31 0.48  0.17 0.46     2.35  0
## [1] " "
## [1] "deltaMinus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad  min  max range skew kurtosis se
## X1    1 342 -0.5  0   -0.5    -0.5   0 -0.5 -0.5     0  NaN      NaN  0
## [1] " "
## [1] "deltaMinus[2,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range  skew
## X1    1 342 -0.54 0.01  -0.54   -0.54 0.01 -0.58 -0.51  0.08 -0.03
##    kurtosis se
## X1     2.41  0
## [1] " "
## [1] "deltaMinus[3,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.57 0.02  -0.57   -0.57 0.01 -0.65 -0.51  0.14  0.1     2.32
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[4,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.56 0.04  -0.53   -0.56 0.02 -0.69 -0.51  0.18 -0.5    -1.16
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[5,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.56 0.02  -0.56   -0.56 0.01 -0.62 -0.51  0.11 0.03     1.94
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[6,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.59 0.02  -0.59   -0.59 0.02 -0.68 -0.52  0.16  0.2      2.2
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[7,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.54 0.01  -0.54   -0.54 0.01 -0.56 -0.51  0.05  0.9     2.09
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad   min   max range skew kurtosis
## X1    1 342 -0.5 0.01   -0.5    -0.5   0 -0.51 -0.46  0.05 1.68     5.33
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[9,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.54 0.01  -0.54   -0.54 0.01 -0.56 -0.51  0.05 0.92     2.46
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[10,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.57 0.02  -0.57   -0.57 0.01 -0.63 -0.51  0.12 0.51     2.36
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.6 0.03   -0.6    -0.6 0.02 -0.69 -0.52  0.17 0.46     2.35
##    se
## X1  0
## [1] " "

2.8.4.1 Create BIC from log likelihood

## extract log likelihood

flare_loglike <- extract_log_lik(flare_fit, parameter_name = "loglik", merge_chains = TRUE)

#calculate BIC

FLARe_bic <- bic(ntrials,colMeans(flare_loglike),3) #number of parameters in that model e.g. 4)

# mean for all participants

mean(FLARe_bic)
## [1] -86.82371

2.8.4.2 Add to bar plot

mod_comp <- rbind(mod_comp,c("Means 2 beta",mean(FLARe_bic)))

plot <- ggplot(na.omit(mod_comp),aes(x=model,y=BIC)) +
  geom_bar(stat = "identity") +
  coord_flip()

show(plot)

2.9 Model 6: RL mode defined,two beta

2.9.1 mode sd RL model with beta per stimulus

RL model adding a beta per stimuli to model defining the beta shape using the mode instead of the mean. This definitely makes more sense as we assume that they will have different levels of uncertainty about each.

## decide testing rate (min,med,max or off)
testing('min')

## set up run
stanname='beta_mode_2beta_RL_2.stan'

stanfile <- file.path(scriptdir, stanname)

flare_data<-list(ntrials=ntrials,nsub=nsub,screamPlus = t(screamPlus), screamMinus= t(screamMinus),ratingsPlus=t(plus_scaled),ratingsMinus=t(minus_scaled))

flare_fit <- stan(file = stanfile, data = flare_data, iter=chain_iter, chains = chain_n) #add working dir?
## 
## SAMPLING FOR MODEL 'beta_mode_2beta_RL_2' NOW (CHAIN 1).
## Chain 1: 
## Chain 1: Gradient evaluation took 0.007511 seconds
## Chain 1: 1000 transitions using 10 leapfrog steps per transition would take 75.11 seconds.
## Chain 1: Adjust your expectations accordingly!
## Chain 1: 
## Chain 1: 
## Chain 1: Iteration:   1 / 400 [  0%]  (Warmup)
## Chain 1: Iteration:  40 / 400 [ 10%]  (Warmup)
## Chain 1: Iteration:  80 / 400 [ 20%]  (Warmup)
## Chain 1: Iteration: 120 / 400 [ 30%]  (Warmup)
## Chain 1: Iteration: 160 / 400 [ 40%]  (Warmup)
## Chain 1: Iteration: 200 / 400 [ 50%]  (Warmup)
## Chain 1: Iteration: 201 / 400 [ 50%]  (Sampling)
## Chain 1: Iteration: 240 / 400 [ 60%]  (Sampling)
## Chain 1: Iteration: 280 / 400 [ 70%]  (Sampling)
## Chain 1: Iteration: 320 / 400 [ 80%]  (Sampling)
## Chain 1: Iteration: 360 / 400 [ 90%]  (Sampling)
## Chain 1: Iteration: 400 / 400 [100%]  (Sampling)
## Chain 1: 
## Chain 1:  Elapsed Time: 57.2036 seconds (Warm-up)
## Chain 1:                32.2233 seconds (Sampling)
## Chain 1:                89.4269 seconds (Total)
## Chain 1:
save(flare_fit, file=file.path(datadir,'flare_fit_test'))

traceplot(flare_fit,'lp__')

# extract fit data
summary_flare <- summary(flare_fit)

# extract model summary data

#flare_loglike<- extract_log_lik(flare_fit, parameter_name = "loglik", merge_chains = TRUE)
## get some basic output descriptions printed to screen

out_describe(summary_flare)
## [1] "400 iterations  on 1 chains. "
## [1] "Estimated 98 Free paramaters per person"
## [1] " "
## [1] "Average Rhat"
## [1] 0.9591849
## [1] "alpha[1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.09 0.03   0.09    0.09 0.02 0.01 0.24  0.24 0.41     2.25  0
## [1] " "
## [1] "beta[1,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342  2.1 2.01   1.55     1.7 0.59 0.84 25.48 24.65 6.22    57.66
##      se
## X1 0.11
## [1] " "
## [1] "beta[172,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 2.13 2.05   1.58    1.73 0.56 0.83 21.48 20.64 5.59    40.12
##      se
## X1 0.11
## [1] " "
## [1] "loglik[1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad    min    max range skew
## X1    1 342 -48.5 4.41 -49.65  -48.97 4.38 -54.31 -34.87 19.44 0.82
##    kurtosis   se
## X1    -0.07 0.24
## [1] " "
## [1] "shape1_Plus[1,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.79 0.47   0.65    0.69 0.18 0.42 4.07  3.65 3.62    16.74
##      se
## X1 0.03
## [1] " "
## [1] "shape1_Plus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min max range skew kurtosis   se
## X1    1 342 0.85 0.48   0.71    0.75 0.2 0.45 4.5  4.05 3.72    18.19 0.03
## [1] " "
## [1] "shape1_Plus[3,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342  0.9 0.5   0.77     0.8 0.22 0.48 4.88  4.39  3.8    19.57
##      se
## X1 0.03
## [1] " "
## [1] "shape1_Plus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.88 0.49   0.73    0.78 0.21 0.45 4.33  3.89 3.51    15.96
##      se
## X1 0.03
## [1] " "
## [1] "shape1_Plus[5,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.87 0.5   0.74    0.77 0.21 0.46 4.73  4.27 3.77    19.01
##      se
## X1 0.03
## [1] " "
## [1] "shape1_Plus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.92 0.51   0.79    0.82 0.22 0.49 5.08  4.59 3.85    20.32
##      se
## X1 0.03
## [1] " "
## [1] "shape1_Plus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min  max range skew kurtosis
## X1    1 342 0.84 0.49   0.71    0.75 0.2 0.45 4.51  4.07  3.7    17.94
##      se
## X1 0.03
## [1] " "
## [1] "shape1_Plus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.78 0.47   0.64    0.68 0.18 0.4 4.03  3.63 3.58    16.26
##      se
## X1 0.03
## [1] " "
## [1] "shape1_Plus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.84 0.49    0.7    0.74 0.19 0.45 4.47  4.03 3.69    17.79
##      se
## X1 0.03
## [1] " "
## [1] "shape1_Plus[10,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.89 0.5   0.76     0.8 0.21 0.48 4.86  4.38 3.78     19.2
##      se
## X1 0.03
## [1] " "
## [1] "shape1_Plus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.94 0.52   0.81    0.84 0.23 0.5 5.19  4.68 3.85    20.45
##      se
## X1 0.03
## [1] " "
## [1] "shape1_Plus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.98 0.54   0.85    0.88 0.24 0.53 5.48  4.95 3.92    21.53
##      se
## X1 0.03
## [1] " "
## [1] "shape1_Minus[1,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.33 1.3   0.94    1.05 0.32 0.48 12.74 12.26 4.87    30.47
##      se
## X1 0.07
## [1] " "
## [1] "shape1_Minus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.45 1.38   1.04    1.15 0.35 0.55 13.22 12.67 4.68    27.91
##      se
## X1 0.07
## [1] " "
## [1] "shape1_Minus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.55 1.44   1.12    1.24 0.37 0.59 13.67 13.07 4.56    26.29
##      se
## X1 0.08
## [1] " "
## [1] "shape1_Minus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342  1.5 1.42   1.05     1.2 0.34 0.51 14.09 13.58 4.79    29.82
##      se
## X1 0.08
## [1] " "
## [1] "shape1_Minus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.49 1.42   1.07    1.19 0.36 0.56 13.54 12.99 4.64    27.43
##      se
## X1 0.08
## [1] " "
## [1] "shape1_Minus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min   max range skew kurtosis
## X1    1 342 1.59 1.48   1.15    1.27 0.39 0.6 13.98 13.38 4.54    26.03
##      se
## X1 0.08
## [1] " "
## [1] "shape1_Minus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.44 1.38   1.03    1.14 0.35 0.53 13.44 12.91 4.77    29.18
##      se
## X1 0.07
## [1] " "
## [1] "shape1_Minus[8,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.31 1.3   0.92    1.04 0.32 0.45 12.93 12.48    5    32.24
##      se
## X1 0.07
## [1] " "
## [1] "shape1_Minus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 1.43 1.38   1.02    1.14 0.35 0.53 13.4 12.87 4.77    29.21
##      se
## X1 0.07
## [1] " "
## [1] "shape1_Minus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.54 1.45   1.11    1.23 0.37 0.58 13.84 13.26 4.62    27.22
##      se
## X1 0.08
## [1] " "
## [1] "shape1_Minus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min   max range skew kurtosis
## X1    1 342 1.63 1.51   1.18     1.3 0.4 0.62 14.26 13.64 4.53    25.95
##      se
## X1 0.08
## [1] " "
## [1] "shape1_Minus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342  1.7 1.57   1.25    1.37 0.42 0.66 14.66    14 4.47    25.12
##      se
## X1 0.08
## [1] " "
## [1] "shape2_Plus[1,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.79 0.47   0.65    0.69 0.18 0.42 4.07  3.65 3.62    16.74
##      se
## X1 0.03
## [1] " "
## [1] "shape2_Plus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.73 0.45    0.6    0.64 0.17 0.37 3.64  3.27 3.54    15.56
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.68 0.44   0.54    0.59 0.17 0.32 3.57  3.25 3.47    14.88
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342  0.7 0.45   0.57    0.61 0.18 0.29 3.8  3.51 3.59    16.59
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342  0.7 0.44   0.57    0.61 0.17 0.34 3.57  3.23 3.51     15.2
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis
## X1    1 342 0.65 0.43   0.52    0.57 0.16 0.3 3.55  3.24 3.46    14.78
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.73 0.44    0.6    0.64 0.17 0.38 3.63  3.25 3.56    15.84
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.79 0.46   0.66     0.7 0.18 0.43 4.11  3.69 3.65    17.16
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.73 0.44    0.6    0.64 0.17 0.38 3.67  3.29 3.57    15.93
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.68 0.43   0.55    0.59 0.17 0.33 3.55  3.22 3.51    15.19
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis
## X1    1 342 0.63 0.42    0.5    0.55 0.16 0.29 3.52  3.23 3.47    14.86
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Plus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range skew kurtosis
## X1    1 342 0.59 0.41   0.46    0.51 0.15 0.26 3.5  3.24 3.45    14.83
##      se
## X1 0.02
## [1] " "
## [1] "shape2_Minus[1,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.33 1.3   0.94    1.05 0.32 0.48 12.74 12.26 4.87    30.47
##      se
## X1 0.07
## [1] " "
## [1] "shape2_Minus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed mad  min   max range skew kurtosis
## X1    1 342 1.21 1.23   0.85    0.96 0.3 0.41 12.27 11.86  5.1    33.44
##      se
## X1 0.07
## [1] " "
## [1] "shape2_Minus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.11 1.18   0.77    0.87 0.28 0.34 11.82 11.48  5.3     35.9
##      se
## X1 0.06
## [1] " "
## [1] "shape2_Minus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.16 1.21   0.82    0.91 0.29 0.26 11.39 11.13 4.89    30.37
##      se
## X1 0.07
## [1] " "
## [1] "shape2_Minus[5,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.17 1.2   0.81    0.92 0.28 0.38 11.94 11.56 5.16    34.24
##      se
## X1 0.06
## [1] " "
## [1] "shape2_Minus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.07 1.15   0.73    0.84 0.26 0.33 11.51 11.18 5.35    36.58
##      se
## X1 0.06
## [1] " "
## [1] "shape2_Minus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.22 1.22   0.85    0.97 0.28 0.43 12.04 11.62 4.99    31.98
##      se
## X1 0.07
## [1] " "
## [1] "shape2_Minus[8,1]"
## [1] " "
##    vars   n mean  sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.35 1.3   0.96    1.07 0.32 0.51 12.55 12.05 4.75    28.75
##      se
## X1 0.07
## [1] " "
## [1] "shape2_Minus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.23 1.23   0.85    0.97 0.29 0.43 12.09 11.66 4.99    31.94
##      se
## X1 0.07
## [1] " "
## [1] "shape2_Minus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.12 1.17   0.78    0.88 0.28 0.37 11.64 11.28  5.2    34.71
##      se
## X1 0.06
## [1] " "
## [1] "shape2_Minus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 1.03 1.12    0.7     0.8 0.26 0.31 11.23 10.92 5.38    36.95
##      se
## X1 0.06
## [1] " "
## [1] "shape2_Minus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min   max range skew kurtosis
## X1    1 342 0.96 1.08   0.65    0.74 0.25 0.24 10.83 10.59 5.53    38.67
##      se
## X1 0.06
## [1] " "
## [1] "VPlus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad min max range skew kurtosis se
## X1    1 342  0.5  0    0.5     0.5   0 0.5 0.5     0  NaN      NaN  0
## [1] " "
## [1] "VPlus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis se
## X1    1 342 0.54 0.02   0.54    0.54 0.01 0.5 0.62  0.12 0.41     2.25  0
## [1] " "
## [1] "VPlus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.58 0.03   0.58    0.58 0.02 0.51 0.71   0.2 0.23     1.84  0
## [1] " "
## [1] "VPlus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.57 0.05   0.53    0.56 0.03 0.51 0.77  0.27 0.77    -0.21  0
## [1] " "
## [1] "VPlus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.56 0.02   0.56    0.56 0.02 0.51 0.64  0.13 0.09      1.2  0
## [1] " "
## [1] "VPlus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342  0.6 0.03    0.6     0.6 0.02 0.51 0.71   0.2 -0.06     1.23
##    se
## X1  0
## [1] " "
## [1] "VPlus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.54 0.01   0.54    0.54 0.01 0.51 0.56  0.05 -0.96     1.57
##    se
## X1  0
## [1] " "
## [1] "VPlus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.49 0.01   0.49    0.49 0.01 0.39 0.51  0.12 -2.76    16.13
##    se
## X1  0
## [1] " "
## [1] "VPlus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.54 0.01   0.54    0.54 0.01 0.51 0.56  0.05 -0.87     1.74
##    se
## X1  0
## [1] " "
## [1] "VPlus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.58 0.02   0.58    0.58 0.02 0.51 0.66  0.15 -0.26     1.44
##    se
## X1  0
## [1] " "
## [1] "VPlus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.61 0.03   0.61    0.61 0.02 0.51 0.74  0.22 -0.2     1.47  0
## [1] " "
## [1] "VPlus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.64 0.04   0.64    0.64 0.03 0.52 0.79  0.28 -0.26     1.43
##    se
## X1  0
## [1] " "
## [1] "VMinus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad min max range skew kurtosis se
## X1    1 342  0.5  0    0.5     0.5   0 0.5 0.5     0  NaN      NaN  0
## [1] " "
## [1] "VMinus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad min  max range skew kurtosis se
## X1    1 342 0.54 0.02   0.54    0.54 0.01 0.5 0.62  0.12 0.41     2.25  0
## [1] " "
## [1] "VMinus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.58 0.03   0.58    0.58 0.02 0.51 0.71   0.2 0.23     1.84  0
## [1] " "
## [1] "VMinus[4,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.57 0.05   0.53    0.56 0.03 0.51 0.77  0.27 0.77    -0.21  0
## [1] " "
## [1] "VMinus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.56 0.02   0.56    0.56 0.02 0.51 0.64  0.13 0.09      1.2  0
## [1] " "
## [1] "VMinus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342  0.6 0.03    0.6     0.6 0.02 0.51 0.71   0.2 -0.06     1.23
##    se
## X1  0
## [1] " "
## [1] "VMinus[7,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.54 0.01   0.54    0.54 0.01 0.51 0.56  0.05 -0.96     1.57
##    se
## X1  0
## [1] " "
## [1] "VMinus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.49 0.01   0.49    0.49 0.01 0.39 0.51  0.12 -2.76    16.13
##    se
## X1  0
## [1] " "
## [1] "VMinus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.54 0.01   0.54    0.54 0.01 0.51 0.56  0.05 -0.87     1.74
##    se
## X1  0
## [1] " "
## [1] "VMinus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.58 0.02   0.58    0.58 0.02 0.51 0.66  0.15 -0.26     1.44
##    se
## X1  0
## [1] " "
## [1] "VMinus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.61 0.03   0.61    0.61 0.02 0.51 0.74  0.22 -0.2     1.47  0
## [1] " "
## [1] "VMinus[12,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.64 0.04   0.64    0.64 0.03 0.52 0.79  0.28 -0.26     1.43
##    se
## X1  0
## [1] " "
## [1] "deltaPlus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad min max range skew kurtosis se
## X1    1 342  0.5  0    0.5     0.5   0 0.5 0.5     0  NaN      NaN  0
## [1] " "
## [1] "deltaPlus[2,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min max range  skew kurtosis se
## X1    1 342 0.46 0.02   0.46    0.46 0.01 0.38 0.5  0.12 -0.41     2.25  0
## [1] " "
## [1] "deltaPlus[3,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad   min  max range skew kurtosis
## X1    1 342 -0.1 0.51  -0.52    -0.1 0.19 -0.69 0.49  1.18 0.06    -1.99
##      se
## X1 0.03
## [1] " "
## [1] "deltaPlus[4,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min  max range  skew kurtosis
## X1    1 342 -0.05 0.54   0.46   -0.04 0.04 -0.77 0.49  1.27 -0.06    -1.99
##      se
## X1 0.03
## [1] " "
## [1] "deltaPlus[5,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range  skew kurtosis
## X1    1 342 0.44 0.02   0.44    0.44 0.02 0.36 0.49  0.13 -0.09      1.2
##    se
## X1  0
## [1] " "
## [1] "deltaPlus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.6 0.03   -0.6    -0.6 0.02 -0.71 -0.51   0.2 0.06     1.23
##    se
## X1  0
## [1] " "
## [1] "deltaPlus[7,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.54 0.01  -0.54   -0.54 0.01 -0.56 -0.51  0.05 0.96     1.57
##    se
## X1  0
## [1] " "
## [1] "deltaPlus[8,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.51 0.01   0.51    0.51 0.01 0.49 0.61  0.12 2.76    16.13  0
## [1] " "
## [1] "deltaPlus[9,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.46 0.01   0.46    0.46 0.01 0.44 0.49  0.05 0.87     1.74  0
## [1] " "
## [1] "deltaPlus[10,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.42 0.02   0.42    0.42 0.02 0.34 0.49  0.15 0.26     1.44  0
## [1] " "
## [1] "deltaPlus[11,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad  min  max range skew kurtosis se
## X1    1 342 0.39 0.03   0.39    0.39 0.02 0.26 0.49  0.22  0.2     1.47  0
## [1] " "
## [1] "deltaMinus[1,1]"
## [1] " "
##    vars   n mean sd median trimmed mad  min  max range skew kurtosis se
## X1    1 342 -0.5  0   -0.5    -0.5   0 -0.5 -0.5     0  NaN      NaN  0
## [1] " "
## [1] "deltaMinus[2,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min  max range  skew kurtosis
## X1    1 342 -0.54 0.02  -0.54   -0.54 0.01 -0.62 -0.5  0.12 -0.41     2.25
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[3,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range  skew
## X1    1 342 -0.58 0.03  -0.58   -0.58 0.02 -0.71 -0.51   0.2 -0.23
##    kurtosis se
## X1     1.84  0
## [1] " "
## [1] "deltaMinus[4,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range  skew
## X1    1 342 -0.57 0.05  -0.53   -0.56 0.03 -0.77 -0.51  0.27 -0.77
##    kurtosis se
## X1    -0.21  0
## [1] " "
## [1] "deltaMinus[5,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range  skew
## X1    1 342 -0.56 0.02  -0.56   -0.56 0.02 -0.64 -0.51  0.13 -0.09
##    kurtosis se
## X1      1.2  0
## [1] " "
## [1] "deltaMinus[6,1]"
## [1] " "
##    vars   n mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.6 0.03   -0.6    -0.6 0.02 -0.71 -0.51   0.2 0.06     1.23
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[7,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.54 0.01  -0.54   -0.54 0.01 -0.56 -0.51  0.05 0.96     1.57
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[8,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.49 0.01  -0.49   -0.49 0.01 -0.51 -0.39  0.12 2.76    16.13
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[9,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.54 0.01  -0.54   -0.54 0.01 -0.56 -0.51  0.05 0.87     1.74
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[10,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.58 0.02  -0.58   -0.58 0.02 -0.66 -0.51  0.15 0.26     1.44
##    se
## X1  0
## [1] " "
## [1] "deltaMinus[11,1]"
## [1] " "
##    vars   n  mean   sd median trimmed  mad   min   max range skew kurtosis
## X1    1 342 -0.61 0.03  -0.61   -0.61 0.02 -0.74 -0.51  0.22  0.2     1.47
##    se
## X1  0
## [1] " "

The alpha parameter variance is normal (mean 0.4 and sd 0.12). Beta is much more bounded now though (combined across both stimuli mean 0.79, sd=1.6) over 4000 iterations on 4 chains.

2.9.1.1 Create BIC from log likelihood

## extract log likelihood

flare_loglike <- extract_log_lik(flare_fit, parameter_name = "loglik", merge_chains = TRUE)

#calculate BIC

FLARe_bic<-bic(ntrials,colMeans(flare_loglike),2) #number of parameters in that model e.g. 4)

## mean BIC as model comparisons tool:

print("Mean Bayesian information criterion for model")
## [1] "Mean Bayesian information criterion for model"
mean(FLARe_bic)
## [1] -92.02652

2.9.1.2 Add to bar plot

mod_comp <- rbind(na.omit(mod_comp),c("Mode 2 beta",mean(FLARe_bic)))

plot <- ggplot(mod_comp,aes(x=model,y=BIC)) +
  geom_bar(stat = "identity") +
  coord_flip()

show(plot)

# to do

  • get means working.
  • add log likelihood to scripts

3 Push any updates to github

3.0.0.0.1 Any push the updates to github

Unhash the series below if you made any changes.


## initialise bash directory and filename
stanname="punish_only.stan"
scriptdir="/Users/kirstin/Dropbox/SGDP/FLARe/FLARe_MASTER/Projects/Hierachal_modelling/Scripts"


## stage
#git add $scriptdir/$stanname


## push 

#git push Bayes_modelling